提取numpy数组中的特定列

Question

这是一个简单的问题，但假设我有一个 MxN 矩阵。我想要做的就是提取特定的列并将它们存储在另一个 numpy 数组中，但我得到无效的语法错误。这是代码：

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

上面的行似乎就足够了，但我想不是。我环顾四周，但找不到任何关于这个特定场景的语法。

原文由 Aladdin 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

我假设您想要列 1 和 9 ？

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

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

要一次选择一个，请使用

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

---

有名字：

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

您可以从 data.dtype.names …

原文由 Fred Foo 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

假设您想使用该代码片段获取第 1 列和第 9 列，它应该是：

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

原文由 Michael J. Barber 发布，翻译遵循 CC BY-SA 3.0 许可协议