n2d = np.array( [ [10,0], [11,0], [12,0], [0,1], [100,1], [200,1], [20,0], [21,0], [22,0], ])
这是x-y坐标的ndarray。我想得到y=0的坐标中的最小和最大x值的坐标我想要得到的值是[10,0],[22,0]我想用NumPy拥有的特性来实现它。
jk9hmnmh1#
import numpy as np n2d = np.array( [ [10,0], [11,0], [12,0], [0,1], [100,1], [200,1], [20,0], [21,0], [22,0], ]) min_y_coordinate = np.amin(n2d) filter_list = [True if item[-1] == min_y_coordinate else False for item in n2d ] filtered_array = n2d[filter_list] print("min_value" , np.amin(filtered_array, axis=0)) print("max_value" , np.amax(filtered_array, axis=0))
1条答案
按热度按时间jk9hmnmh1#