读取csv文件并写成list
时间: 2023-09-01 09:03:56 浏览: 89
读取csv文件并将其写入列表可以使用Python的csv模块来实现。以下是实现的代码:
import csv
def read_csv(file_path):
data = []
with open(file_path, 'r') as file:
csv_reader = csv.reader(file)
for row in csv_reader:
data.append(row)
return data
file_path = 'example.csv'
csv_data = read_csv(file_path)
print(csv_data)
代码解释:
1. 首先我们导入csv模块。
2. 然后定义一个`read_csv`函数,它接受一个csv文件路径作为参数。
3. 在函数中,我们创建一个空列表`data`,用于存储读取到的csv数据。
4. 使用`open`函数打开csv文件,模式为只读模式('r')。
5. 创建一个csv_reader对象,该对象使用`csv.reader`函数来读取csv文件。
6. 使用for循环遍历csv_reader对象,将每一行数据追加到`data`列表中。
7. 循环结束后,关闭文件。
8. 返回列表`data`。
9. 在主程序中,定义一个csv文件路径`file_path`(此处假设文件名为example.csv)。
10. 调用`read_csv`函数,并将返回的数据赋值给变量`csv_data`。
11. 最后,打印`csv_data`列表。
上述代码将会读取example.csv文件的内容,并将其写入一个二维列表(即列表的列表),其中每个子列表表示csv文件中的一行数据。假设example.csv的内容如下所示:
1,John,Doe
2,Jane,Smith
3,David,Johnson
运行代码后,输出结果为:
[['1', 'John', 'Doe'], ['2', 'Jane', 'Smith'], ['3', 'David', 'Johnson']]
阅读全文