ndarray存和读csv

在Python中，可以使用NumPy库来创建和操作多维数组（ndarray）。如果我们有一个ndarray，想要将其存储为CSV文件，可以使用NumPy的`savetxt()`函数。如果我们想要从CSV文件中读取数据并将其加载到ndarray中，可以使用NumPy的`loadtxt()`函数。

下面是一个示例代码，演示了如何将ndarray存储为CSV文件并从CSV文件中读取数据到ndarray中：

import numpy as np

# 创建一个2维ndarray
data = np.array([[1, 2, 3], [4, 5, 6]])

# 将ndarray存储为CSV文件
np.savetxt('data.csv', data, delimiter=',')

# 从CSV文件中读取数据到ndarray中
loaded_data = np.loadtxt('data.csv', delimiter=',')

print("原始数据：")
print(data)

print("从CSV文件中读取的数据：")
print(loaded_data)

输出结果如下：

原始数据：
[[1 2 3]
 [4 5 6]]
 
从CSV文件中读取的数据：
[[1. 2. 3.]
 [4. 5. 6.]]

注意，从CSV文件中读取的数据类型为float类型，因为NumPy的`loadtxt()`函数默认将所有数据读取为float类型。如果需要指定数据类型，可以使用`dtype`参数。例如，如果我们想要从CSV文件中读取整数数据，可以将`dtype`参数设置为整数类型，如下所示：

loaded_data = np.loadtxt('data.csv', delimiter=',', dtype=int)