在numpy数组中的每一行应用函数?

时间:2017-08-14 03:00:50

标签: python arrays 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"获得的数组的每一行。进入"定期"行得到结果。如何以一种我不需要将数组的行放入"周期性"赞扬?

谢谢。

2 个答案:

答案 0 :(得分:2)

你好运!有一个很棒的东西叫做for循环,它可以让你依次循环遍历每一行并计算它们。

fra = struct.frac_coords

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

答案 1 :(得分:0)

我使用list comprehension来完成这项工作。

{{1}}