要编写一个类似于CSV文件的文件,可以使用Python的csv模块。下面是一个示例代码:
import csv
def write_csv_file(data, filename):
with open(filename, 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
writer.writerows(data)
def read_csv_file(filename):
with open(filename, 'r') as csvfile:
reader = csv.reader(csvfile)
data = [row for row in reader]
return data
# 示例数据
data = [
['Name', 'Age', 'Country'],
['John', '25', 'USA'],
['Alex', '30', 'UK'],
['Lisa', '28', 'Canada']
]
# 写入CSV文件
write_csv_file(data, 'example.csv')
# 读取CSV文件
data = read_csv_file('example.csv')
print(data)
上述代码中,write_csv_file
函数用于将数据写入CSV文件,read_csv_file
函数用于从CSV文件中读取数据。
注意,csv.writer
和csv.reader
类都需要一个文件对象作为参数,因此需要使用open
函数打开一个文件。在写入文件时,需要将文件的打开模式设置为'w'
,表示以写入模式打开文件;在读取文件时,使用默认的读取模式'r'
即可。
运行以上代码后,会在当前目录下生成一个名为example.csv
的文件,并且成功读取文件中的数据并打印出来。