我已经尽力找到自己的解决方案但是我没有提出任何相关的解决方案。我有一些从.tbl文件中提取的numpy数组(这是我用atpy提取的天文表格式)。每个阵列中有大约25,000个数字,我将它们绘制为带有matplotlib的散点图。 y轴值直接来自数组,x轴是两个独立数组中值的简单减法。
这一切都很好,但我真正需要做的是提取落在一定范围内的值(例如,我需要y介于10和13之间的值,x的值介于0和1之间)。当然,要使情节起作用,这些价值观必须相互匹配。这就是我所拥有的:
import numpy as np
from numpy import ndarray
import matplotlib.pyplot as plt
import matplotlib
import atpy
twomass = atpy.Table()
twomass.read('/Raw_Data/IRSA_downloads/2MASS_GCbox1.tbl')
hmag = list([twomass['h_m']])
jmag = list([twomass['j_m']])
colorjh = list([j-h for j,h in zip(jmag, hmag)])
x = []
y = []
def list_maker():
for c,h in zip(colorjh, hmag):
if np.all(c) < 1 == True and np.all(c) > 0 == True and np.all(h) < 13 == True and np.all(h) > 10 == True:
x.append(c)
y.append(h)
list_maker()
plt.scatter(x, y, c='g', s=1, alpha=0.05)
plt.xlabel('Color(J-H)', fontsize=15) #adjust axis labels here
plt.ylabel('Magnitude (H)', fontsize=15)
plt.gca().invert_yaxis()
plt.legend(loc=2)
plt.title('CMD for Galactic Center (2MASS)', fontsize=20)
plt.grid(True)
plt.show()
我也尝试过这种方法来切片数据,但没有运气:
colorjh = colorjh[colorjh<1]
colorjh = colorjh[colorjh>0]
我已经尝试了很多不同的东西,包括尝试将这些数组转换为列表,以及不平等的格式化的许多不同的东西。在这个过程中,我可能已经离开了答案,但这段代码至少打印出整个分散(它无法像我想的那样切断数据)。其他一些迭代打印的空白图不在我正在寻找的数字范围内。
我是python和这个网站的新手,所以请尽可能明确地提出任何提示,等等。如果这是超级jargony我可能无法很好地利用你的建议。感谢大家。
答案 0 :(得分:7)
请尝试以下操作,我认为它与您的for
和zip
相同,但速度要快得多。如果某些事情没有意义,请在评论中提问,我将编辑:
hmag = np.array(hmag)
jmag = np.array(jmah)
colorjh = jmag - hmag
idx_c = (colorjh > 0) & (colorjh < 1) # where condition on c is met
idx_h = (hmag > 10) & (hmag < 13) # where condition on h is met
idx = idx_c & idx_h # where both conditions are met
plt.scatter(colorjh[idx], hmag[idx], c='g', s=1, alpha=0.05)
答案 1 :(得分:2)
这两个条件都可以立即完成:
hmag = np.array(hmag)
jmag = np.array(jmah)
colorjh = jmag - hmag
idx = ((colorjh > 0) & (colorjh < 1) & (hmag > 10) & (hmag < 13)).nonzero()[0]
plt.scatter(colorjh[idx], hmag[idx], c='g', s=1, alpha=0.05)
.nonzero()[0]
使得它成为索引列表,而不是具有True和False值的'mask',如果我们讨论非常长的列表,这可能会更有效。