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

