提取numpy数组中的特定列
提取numpy数组中的特定列
提问于 2011-12-05 22:20:48
这是一个简单的问题，但是假设我有一个MxN矩阵。我所要做的就是提取特定的列，并将它们存储在另一个numpy数组中，但是我得到了无效的语法错误。代码如下：
extractedData = data[[:,1],[:,9]]. 上面这行看起来应该足够了，但我猜不够。我环顾四周，但找不到关于这个特定场景的任何语法方面的东西。
回答 7
Stack Overflow用户
回答已采纳
发布于 2011-12-05 22:24:54
我假设您想要列1和9
要一次选择多个列，请使用
X = data[:, [1, 9]]若要一次选择一个，请使用
x, y = data[:, 1], data[:, 9]使用名称：
data[:, ['Column Name1','Column Name2']]您可以从data.dtype.names…获取名称
Stack Overflow用户
发布于 2011-12-05 22:26:02
假设您想使用该代码片段获取第1列和第9列，它应该是：
extractedData = data[:,[1,9]]Stack Overflow用户
发布于 2015-06-01 19:14:00
如果只想提取一些列：
idx_IN_columns = [1, 9]
extractedData = data[:,idx_IN_columns]如果要排除特定列：
idx_OUT_columns = [1, 9]
idx_IN_columns = [i for i in xrange(np.shape(data)[1]) if i not in idx_OUT_columns]
extractedData = data[:,idx_IN_columns]页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/8386675
复制相关文章
相似问题