繁体   English   中英

对 numpy 数组做一些数学运算

[英]Do some mathematical operation on numpy array

我有一个 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 进行比较,如果坐标在范围内,那么它们将 append 否则我们创建新数组。

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


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: invalid index and scalar variable

我不确定我是否理解这个问题,但只是为了让事情开始:

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])}')

output:

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) 

output:

[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]

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM