python - 将函数应用于numpy数组中的每一行？

我正在尝试编写包含以下行的脚本，

fra = struct.frac_coords

和“ fra”给出以下答案：

array([[ 0.01106406,  0.11554355,  0.50754736],
   [ 0.00294858,  0.24640931,  0.99887037],
   [ 0.37046412,  0.08121749,  0.9386742 ],
   [ 0.49430774,  0.07065645,  0.7479905 ],
   [ 0.6249222 ,  0.04073112,  0.56187813]])

然后我有以下一行：

periodic = struct.get_sites([ 0.01106406,  0.11554355,  0.50754736], 5.95, include_index=True)

print(periodic)

因此，我必须将从“ fra”获得的数组的每一行输入“ periodic”行中以获得结果。如何以不需要将数组的线条放入“定期”推荐中的方式制作线条？

谢谢。

最佳答案

你真幸运！有一个很棒的事情叫做for循环，它使您可以依次循环遍历每行并进行计算。

fra = struct.frac_coords

for row in fra:
    periodic = struct.get_sites(row, 5.95, include_index=True)
    print(periodic)

关于python - 将函数应用于numpy数组中的每一行？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/45666858/