用python csv写文件,输出到指定文件夹,用pandas打开
时间: 2024-03-09 15:48:21 浏览: 219
使用Python(pandas库)处理csv数据
你可以使用 Python 的 csv 模块来写入 CSV 文件,并使用 pandas 库来打开 CSV 文件。
下面是一个示例代码,假设你要写入的数据为一个列表 of 列表:
```python
import csv
import os
import pandas as pd
# 定义数据
data = [[1, 'apple', 0.5],
[2, 'banana', 0.3],
[3, 'pear', 0.6]]
# 定义输出路径
output_dir = 'output'
if not os.path.exists(output_dir):
os.makedirs(output_dir)
output_file = os.path.join(output_dir, 'output.csv')
# 写入 CSV 文件
with open(output_file, 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
writer.writerow(['id', 'name', 'price'])
writer.writerows(data)
# 读取 CSV 文件
df = pd.read_csv(output_file)
print(df)
```
代码中的第一部分定义了要写入的数据,是一个列表 of 列表。第二部分定义了输出路径,这里将输出文件存储在 `output` 文件夹下,如果该文件夹不存在,则会创建该文件夹。第三部分使用 `csv` 模块将数据写入 CSV 文件。`with open()` 语句打开文件,`csv.writer` 创建一个写入器,`writerow` 写入表头(例如 "id", "name", "price"),`writerows` 写入数据。最后使用 pandas 库的 `read_csv` 函数读取 CSV 文件,并打印出来。
注意,由于写入 CSV 文件时使用了 `newline=''` 参数,所以在 Windows 系统上运行时,可能需要将文件以二进制形式打开(即使用 `'wb'` 模式)。
阅读全文