一尘不染

提取numpy数组中的特定列

python

这是一个简单的问题,但请说我有一个MxN矩阵。我要做的就是提取特定的列并将其存储在另一个numpy数组中,但是我得到了无效的语法错误。这是代码:

extractedData = data[[:,1],[:,9]].

似乎上述行就足够了,但我想不是。我环顾四周,但找不到关于此特定场景的任何语法明智的方法。


阅读 157

收藏
2020-12-20

共1个答案

一尘不染

我假设你想要的列19

要一次选择多个列,请使用

X = data[:, [1, 9]]

要一次选择一个,请使用

x, y = data[:, 1], data[:, 9]

带有名称:

data[:, ['Column Name1','Column Name2']]

您可以从data.dtype.names…获得名字。

2020-12-20