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

Question

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

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”行中才能得到结果。如何以不需要将数组的行放入“周期性”命令的方式来制作行？

谢谢。

Answer 1

你很幸运!有一个很棒的东西叫做 for循环，让您依次循环每一行并计算相同的值。

fra = struct.frac_coords

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