当前位置:首页python > 正文

csv文件存储二维数组的例子

作者:野牛程序员:2023-07-11 19:23:56python阅读 2875

以下是一个将二维数组存储为CSV文件的Python示例:

import csv

def write_csv(filename, data):
    with open(filename, 'w', newline='') as csvfile:
        writer = csv.writer(csvfile)
        for row in data:
            writer.writerow(row)

def read_csv(filename):
    data = []
    with open(filename, 'r') as csvfile:
        reader = csv.reader(csvfile)
        for row in reader:
            data.append(row)
    return data

# 二维数组示例数据
array = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

# 将二维数组写入CSV文件
write_csv('data.csv', array)

# 从CSV文件读取二维数组
data = read_csv('data.csv')

print(data)

这个例子中,我们使用Python的csv模块来处理CSV文件。首先,write_csv函数接收一个文件名和一个二维数组作为输入,将数组的每一行写入到CSV文件中。read_csv函数接收一个文件名作为输入,并从CSV文件中读取数据,并返回一个包含所有行的二维数组。

在示例中,我们使用一个包含3行3列的二维数组作为示例数据,将其写入名为"data.csv"的CSV文件中。然后,我们再次从CSV文件中读取数据,并将其存储在变量data中。最后,我们打印出读取的二维数组。

请确保在运行代码之前已安装Python的csv模块。你可以使用以下命令安装它:

pip install csv

运行代码后,你应该能够看到打印出的二维数组与示例数据相同。你也可以打开"data.csv"文件,查看其中的内容。

野牛程序员教少儿编程与信息学奥赛-微信|电话:15892516892
野牛程序员教少儿编程与信息学竞赛-微信|电话:15892516892
相关推荐

最新推荐

热门点击