Python中如何正确获取二维数组的第N列数据

你使用的是numpy多维数组，numpy的索引规则为[行维度, 列维度]，仅传入单个索引时默认访问第一个维度（行维度），因此y[N]会返回第N行。

要正确获取第N列，请使用以下写法：

# 注意numpy索引从0开始计数，N为目标列对应的索引值
y[:, N]

补充说明：

• 写法中的冒号:表示选中当前维度的所有元素，即选中全部行后取对应列的元素，返回结果是长度为9976的一维数组
• 如果需要保留二维列向量格式（形状为(9976, 1)），可以改用y[:, N:N+1]
• 示例：要取数组的第1列，对应索引为0，写法为y[:, 0]

内容的提问来源于stack exchange，提问作者asmgx