função sobre a lista mathematica

Question

Eu trabalho em um problema que precisa definir a função com duas listas de dados como segue

a = {0.1, 0.233, 0.65, 1.56};
b = {0.99, 0.8, 0.9, 1.9};

ELCDM1[z_?NumericQ, om_?NumericQ] := 1/Sqrt[om*(1 + z)^3 + 1 - om];

angdist[zmin_?NumberQ, zmax_?NumberQ, om_?NumberQ] := (3*10^5)/(

   1 + zmax) NIntegrate[ELCDM1[z, om], {z, zmin, zmax}];

"a" é a lista do limite inferior de integrar e b é o limite superior. O verdadeiro resultado para minha função "angdis" é {100707., 71762., 25625.6, 13508.}. Para mim, devo colocar a primeira lista "a" na minha função e, no final do meu programa, colocar a lista "b", de modo que, conforme fiz o seguinte, o resultado será diferente com {100707., 71762., 25625.6, 13508.}

angdist3[x_, om_] := (3*10^5)/(1 + #) NIntegrate[ELCDM1[z, om], {z, x, #}] & /@b;

angdist3[#, 0.3] & /@ a

aqui está o resultado:

{{100707., 92199.9, 97142.2, 111441.}, {82220.6, 71762., 77779.9, 
  98755.}, {32424.9, 16710.2, 25625.6, 
  64584.9}, {-42008.8, -65580.4, -52333.9, 13508.}}

Quero definir a função como angdist3 [x_, om_] no final e fornecer: {100707., 71762., 25625.6, 13508.}, mas fornecer uma lista errada.

Qualquer sugestão seria muito apreciada.

Answer 1

Se você precisar calcular todos os 16 (4 × 4) integrais e extrair os elementos diagonais, use Diagonal:

Diagonal[angdist3[#, 0.3] & /@ a]

{100707., 71762., 25625.6, 13508.}

Em vez disso, você pode calcular apenas esses 4 resultados. MapThreadirá mapear ambos ae bsimultaneamente:

MapThread[angdist[#1, #2, 0.3] &, {a, b}]

{100707., 71762., 25625.6, 13508.}

Como alternativa, você pode fornecer angdisto Listableatributo e simplesmente passar ae bcomo argumentos:

SetAttributes[angdist, Listable]
angdist[a, b, 0.3]

{100707., 71762., 25625.6, 13508.}