结合numpy数组和掩蔽

Question

我有以下问题。在know that numpy中，我可以用genfromtext读取数据文件。然后，我可以用掩码（从字典中）从中选择记录。在

>>> gln = a[a['AA2'] == 'GLN']
>>> gln['mean'].mean()
62.805000305175781

如何使用阵列组合而不是从磁盘读取阵列？在

^{pr2}$

我怎么才能让它工作？在

Answer 1

请记住，在Python中，[]生成的对象通常称为list，而不是array。考虑到这一点，我假设您真正的问题是“如何将4个独立的python lists转换为二维numpy数组？”在

答案取决于您希望list在数组中的布局方式。假设每个列表代表一行，则可以执行以下操作：

>>> x = range(0, 4)
>>> y = range(4, 8)
>>> ov = range(8, 12)
>>> aa = range(12, 16)
>>> numpy.array([x, y, ov, aa])
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [12, 13, 14, 15]])

或者，如果每个列表代表一列，则可以执行以下操作：

^{pr2}$

然后，您可以像使用genfromtxt生成的array一样使用布尔索引：

>>> a = numpy.array(zip(x, y, ov, aa))
>>> a[a == 1]
array([ 1,  5,  9, 13,  3,  7, 11, 15])

可能是您正在寻找其他方法将x、y、ov、和{}组合成一个数组。在这种情况下，您必须指定列表中的内容以及所需输出的形状。在