用python csv写文件,一定要输出到当前文件夹,用pandas打开这个csv
时间: 2023-12-07 07:02:55 浏览: 162
你可以使用 Python 的 csv 模块来写入 CSV 文件,并使用 pandas 库来打开 CSV 文件。
下面是一个示例代码,假设你要写入的数据为一个列表 of 列表:
```python
import csv
import pandas as pd
# 定义数据
data = [[1, 'apple', 0.5],
[2, 'banana', 0.3],
[3, 'pear', 0.6]]
# 定义输出文件名
output_file = '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 列表。第二部分定义了输出文件名,这里将输出文件存储在当前文件夹下。第三部分使用 `csv` 模块将数据写入 CSV 文件。`with open()` 语句打开文件,`csv.writer` 创建一个写入器,`writerow` 写入表头(例如 "id", "name", "price"),`writerows` 写入数据。最后使用 pandas 库的 `read_csv` 函数读取 CSV 文件,并打印出来。
注意,由于写入 CSV 文件时使用了 `newline=''` 参数,所以在 Windows 系统上运行时,可能需要将文件以二进制形式打开(即使用 `'wb'` 模式)。
阅读全文