当前位置:首页python > 正文

Python读写csv或txt文件存入二维数组

作者:野牛程序员:2023-06-27 14:09:26python阅读 2728

在Python中,你可以使用内置的csv模块来读写CSV文件,也可以使用内置的open函数来读写TXT文件。下面是分别读写CSV文件和TXT文件并将数据存入二维数组的示例代码:

读取CSV文件并存入二维数组:

import csv

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

# 示例用法
filename = 'data.csv'
csv_data = read_csv(filename)
print(csv_data)

写入CSV文件:

import csv

def write_csv(filename, data):
    with open(filename, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerows(data)

# 示例用法
filename = 'output.csv'
data = [['Name', 'Age', 'Gender'], ['John', '25', 'Male'], ['Lisa', '30', 'Female']]
write_csv(filename, data)

读取TXT文件并存入二维数组:

def read_txt(filename):
    data = []
    with open(filename, 'r') as file:
        for line in file:
            row = line.strip().split()
            data.append(row)
    return data

# 示例用法
filename = 'data.txt'
txt_data = read_txt(filename)
print(txt_data)

写入TXT文件:

def write_txt(filename, data):
    with open(filename, 'w') as file:
        for row in data:
            line = ' '.join(row) + '\\n'
            file.write(line)

# 示例用法
filename = 'output.txt'
data = [['Name', 'Age', 'Gender'], ['John', '25', 'Male'], ['Lisa', '30', 'Female']]
write_txt(filename, data)

请确保你已经将要读取的CSV或TXT文件放在与Python脚本相同的目录下,并提供正确的文件名。以上代码将读取文件的每一行作为二维数组的一行,并将其存储在名为data的列表中。

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

最新推荐

热门点击