Numpy文件操作

读入文件

import numpy as np

# 读入csv文件或txt文件
data = np.loadtxt("./data.csv", skiprows=5, delimiter=",", max_rows=5)  # 返回数组
# skeprows跳过前5行，delimiter分隔符，max_rows读入最多5行
print(data)

保存文件

import numpy as np

# 保存为csv文件或txt文件
data = np.arange(256).reshape((-1, 2))  # 指定ndarray为两列，行数自动计算
np.savetxt("./data.csv", data, fmt="%.2f", delimiter=",")
# fmt数据格式，%d为整数，%.2f为两位小数，%.2e为带两位小数的10幂次方