Python如何从文件加载数据并转换为形状为(3,3)的数组A

你当前写的代码拿不到3×3数组的原因很简单：末尾的[:, 0]是数组索引操作，只会截取加载结果的第一列，最终返回形状为(3,)的一维数组，完全不符合要求。

直接删掉多余的列索引，同时给np.loadtxt指定CSV文件对应的逗号分隔符，就能直接得到形状为(3,3)的目标数组A，代码如下：

import numpy as np
A = np.loadtxt('A.csv', delimiter=',')

如果你的CSV文件第一行是列名、表头等非数值内容，加个skiprows参数跳过对应行就行，比如第一行是表头的写法：

A = np.loadtxt('A.csv', delimiter=',', skiprows=1)

加载完成后可以跑一句print(A.shape)做校验，输出为(3, 3)就说明形状符合要求，数组结构和你给出的目标结果一致。

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