查找y的最大值和最小值坐标两个3D numpy数组

Question

我是Python的完整入门者，正在尝试使用它来生成要在分子模拟中使用的合并数据文件。

我正在尝试使用ASE python代码将一组原子沿y轴放置在另一组原子上。为此，我需要找到存储为3D numpy数组（atoms_1）的一个原子组的y坐标的最小值和存储为3D numpy数组（atoms_2）的第二个原子组的y坐标的最大值。哪种方法最好呢？

我尝试过：

x = np.max（[atoms_2 [：，1]]）

y = np.min（[atoms_1 [：，1]]）

不确定语法是否正确。任何帮助或提示表示赞赏。

Answer 1

您正在做什么似乎是正确的。您也可以删除括号：

y_min = np.max(atoms_2[:,1])

y_max = np.min(atoms_1[:,1])

np.max是注释中建议的np.amax的别名。

Answer 2

您可以使用以下方法在每列中获得最大值：

import numpy as np

atoms = np.array([[1, 2, 3],
                  [4, 5, 6],
                  [7, 8, 9],
                  [1, 4, 6]])

print(np.amax(atoms, axis=0))

>>> [7 8 9]
print(np.amin(atoms, axis=0))
>>> [1 2 3]

要分配它，只需使用：

x_max, y_max, z_max = np.amax(atoms, axis=0)
x_min, y_min, z_min = np.amin(atoms, axis=0)

Answer 3

您也可以将轴传递到max和min

a = np.arange(27).reshape(3,3,-1)

a.max(axis=(0,2)), a.min(axis=(0,2))

输出：

(array([20, 23, 26]), array([0, 3, 6]))