对numpy数组做一些数学运算

时间:2019-10-14 12:38:18

标签: arrays python-3.x numpy

我有一个numpy数组,如下所示:

 [array([580, 201]), array([167, 701]), array([167, 694]), array([979, 725]), array([ 200, 1271]), array([1005, 1266]), array([ 180, 1568]), array([ 994, 1591]), array([ 539, 1862])]

,我想从和201中提取580,以便可以找到x和y的最小值和最大值。因此,我将其他坐标与x_min,x_max,y_min和y_max进行比较,如果坐标在范围内,则它们将追加,否则我们将创建新数组。

我试图找到提取数组元素,但后来找不到任何对它进行操作的解决方案。


coordinates = [array([580, 201]), array([167, 701]), array([167, 694]), array([979, 725]), array([ 200, 1271]), array([1005, 1266]), array([ 180, 1568]), array([ 994, 1591]), array([ 539, 1862])]

for i in coordinates_new[0]:
    print(i)


output: 580 
201

我试图只提取580和201,所以当我使用i [0]时,我将其与其他坐标进行了比较,这会产生IndexError:无效的索引和标量变量的错误

1 个答案:

答案 0 :(得分:0)

我不确定我是否理解这个问题,只是为了开始工作:

import numpy as np
aa =[[580, 201], [167, 701], [167, 694], [979, 725], [ 200, 1271], [1005, 1266], [ 180, 1568], 
     [ 994, 1591], [ 539, 1862]]
print(f' list :\n{aa}\n')

bb=np.array(aa)
print(f' np.array :\n {bb}\n')

cc=np.transpose(bb)
print(f' transposed array :\n {cc}\n')

print(f' xmax: {np.max(cc[0])}')
print(f' ymax: {np.max(cc[1])}')
print(f' xmin: {np.min(cc[0])}')
print(f' ymin: {np.min(cc[1])}')

输出:

list :
[[580, 201], [167, 701], [167, 694], [979, 725], [200, 1271], [1005, 1266], [180, 1568], [994, 1591], [539, 1862]]

 np.array :
 [[ 580  201]
 [ 167  701]
 [ 167  694]
 [ 979  725]
 [ 200 1271]
 [1005 1266]
 [ 180 1568]
 [ 994 1591]
 [ 539 1862]]

 transposed array :
 [[ 580  167  167  979  200 1005  180  994  539]
 [ 201  701  694  725 1271 1266 1568 1591 1862]]

 xmax: 1005
 ymax: 1862
 xmin: 167
 ymin: 201

对数组的元素应用一些条件:

bb=[]  # empty list
for a in aa:
    print(a)
    if a[0]>200: bb.append(a)   # you can have two conditions, of course, a[1]>..
print()
for b in bb: 
    print(b) 

输出:

[580, 201]
[167, 701]
[167, 694]
[979, 725]
[200, 1271]
[1005, 1266]
[180, 1568]
[994, 1591]
[539, 1862]

[580, 201]
[979, 725]
[1005, 1266]
[994, 1591]
[539, 1862]