función sobre lista matemática

Question

Trabajo en un problema que necesita definir la función con dos listas de datos de la siguiente manera

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" es la lista del límite inferior de la integración y b es el límite superior. El verdadero resultado de mi función "angdis" es {100707., 71762., 25625.6, 13508.}. Para mí, debería poner la primera lista "a" en mi función y al final de mi programa poner la lista "b" para que, como hice a continuación, el resultado sea diferente con {100707., 71762., 25625.6, 13508.}.

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

angdist3[#, 0.3] & /@ a

aquí está el 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.}}

Quiero definir una función como angdist3 [x_, om_] al final y darme: {100707., 71762., 25625.6, 13508.} pero dame una lista incorrecta.

Cualquier sugerencia será muy apreciada.

Answer 1

Si necesita calcular las 16 integrales (4 × 4) y luego extraer los elementos diagonales, use Diagonal:

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

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

En su lugar, podría calcular solo esos 4 resultados. MapThreadmapeará sobre ambos ay bsimultáneamente:

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

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

Alternativamente, puede dar angdistel Listableatributo y simplemente pasar ay bcomo argumentos:

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

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