Python数组保存到文件

2023-11-14 13:34:58

要将Python数组保存到文件中,可以使用以下几种方法:

方法1:使用pickle模块

import pickle

# 定义数组
my_array = [1, 2, 3, 4, 5]

# 将数组保存到文件
with open('array.pickle', 'wb') as file:
    pickle.dump(my_array, file)

方法2:使用numpy库

import numpy as np

# 定义数组
my_array = np.array([1, 2, 3, 4, 5])

# 将数组保存到文件
np.save('array.npy', my_array)

方法3:使用csv模块

import csv

# 定义数组
my_array = [1, 2, 3, 4, 5]

# 将数组保存到文件
with open('array.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(my_array)

方法4:使用pandas库

import pandas as pd

# 定义数组
my_array = [1, 2, 3, 4, 5]

# 创建DataFrame对象
df = pd.DataFrame(my_array)

# 将DataFrame保存到文件
df.to_csv('array.csv', index=False)

这些方法可以将Python数组以不同的格式保存到文件中,根据需要选择适合的方法。