如何在python中使用两个不同的条件来过滤数组
如何在python中使用两个不同的条件来过滤数组
提问于 2018-07-31 05:55:20
我正在尝试创建一个新的数组，它将由来自第一列的(数据值<=20000)以及所有其他相应的列组成。
从数学上讲，我是这样做的:我从一个文本文件中读取数据。我正在计算从最后一个点到所有点的距离。然后，我将只取那些距离小于20000且M1小于11.5时的行。数据如下所示：
# ID M1 M2 M3 M4 R4 M5 R5 x y z
10217 11.467 11.502 13.428 13.599 432.17 13.266 281.06 34972.8 42985.9 14906
7991 11.529 11.559 13.438 13.520 435.23 13.224 272.23 8538.05 33219.8 43375.1
2100 11.526 11.573 13.478 13.490 448.97 13.356 301.27 9371.75 13734.1 43398.6
9467 11.557 11.621 13.481 13.537 449.99 13.367 303.67 33200.3 36008.9 12735.8我的代码如下所示：
import numpy as np
import matplotlib.pyplot as plt
halo = 'nntest.txt'
ID, m,r,x,y,z= np.loadtxt(halo, usecols=(0,6,7,8,9,10), unpack =True)
# selet the last point
m_mass = m[-1:]
ID_2 = ID[-1:]
r_2 = r[-1:]
x_2 = x[-1:]
y_2 = y[-1:]
z_2 = z[-1:]
#######################################
#Find distance to all points from our targeted point
nearest_neighbors = []
def neighbors(ID_2, cx,cy,cz, ID, m, r, x, y, z):
dist = np.sqrt((cx-x)**2 + (cy-y)**2 + (cz-z)**2)
return dist, ID, m, r, x, y, z
for i in range(len(ID_2)):
hist = neighbors(ID_2[i], m_2[i], r_2[i], x_2[i], y_2[i], z_2[i], ID, m , r, x, y, z)
#print all the IDs and all other data which are less than 20000 and M less than 11.5 of that targeted value
print ID[hist[0]<20000] and m[hist[1]<11.5]但我在设置这两个条件时遇到了问题。它返回这个错误：
File "overlaping_halos_finder.py", line 53, in <module>
combined = zip(ID[hist[0]<r_2[i] and m[hist[1]>1.e12]])
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()如果不是
print ID[hist[0]<20000] and m[hist[1]<11.5]我只做：
print ID[hist[0]<20000]我有以下示例输出：
# ID M R X Y Z
6737.0 909913272632.0 103.06 1988.35 15894.6 40649.0
6617.0 997700063823.0 106.28 1523.55 15433.2 40363.2
6723.0 11 109.91 1993.05 15687.5 40557.2但我想去掉前两个输出，只打印M列值<11.5的第3行
希望它能澄清一点
您对解决此问题的建议将非常有帮助
回答 2
Stack Overflow用户
回答已采纳
发布于 2018-07-31 06:43:01
您不需要逻辑and。要打印数组的tuple，只需打印逗号分隔的数组序列：
print ID[hist[0]<20000], m[hist[1]<11.5]ID将由布尔数组hist[0]<20000索引，而m将由布尔数组hist[1]<11.5索引。
如果你需要赋值给变量，你可以使用序列解包：
res1, res2 = ID[hist[0]<20000], m[hist[1]<11.5]Stack Overflow用户
发布于 2018-07-31 06:12:23
我不理解你的代码，但从标题中我建议你定义一个函数，它给出一个点，如果应该过滤该点，则返回true。然后您可以执行以下操作：
def filter_point(point):
# Conditions over a single point
# You can read from global variables here, or create a partial function
# to access external data
return condition1(point) and condition2(point)
result = filter(filter_point, list_of_points)页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/51602799
复制相关文章
相似问题