文件路径不确定,用python的csv写入文件后,再读取并用pandas读取并画图。写出代码
时间: 2023-07-18 12:30:14 浏览: 121
使用python的pandas库读取csv文件保存至mysql数据库
如果文件路径不确定,你可以使用Python的os模块来获取当前工作目录,并将文件名添加到路径中。下面是一个示例代码:
```python
import csv
import pandas as pd
import matplotlib.pyplot as plt
import os
# 获取当前工作目录并拼接文件名
filename = 'output.csv'
path = os.path.join(os.getcwd(), filename)
# 写入CSV文件
with open(path, 'w', newline='') as file:
writer = csv.writer(file)
writer.writerow(['Name', 'Age', 'Gender'])
writer.writerow(['Alice', 25, 'Female'])
writer.writerow(['Bob', 30, 'Male'])
writer.writerow(['Charlie', 35, 'Male'])
# 读取CSV文件
df = pd.read_csv(path)
# 绘制柱状图
df.plot(kind='bar', x='Name', y='Age')
plt.show()
```
在这个例子中,我们使用os模块中的getcwd()函数获取当前工作目录,并使用join()函数将文件名添加到路径中。然后,我们使用CSV库的writer对象将数据写入到文件中,并使用Pandas库的read_csv()函数读取该文件。最后,我们使用DataFrame对象的plot()方法绘制柱状图,并使用Matplotlib库的show()函数显示图表。
请注意,在使用os模块拼接路径时,需要注意操作系统的文件路径分隔符。在Windows系统中,路径分隔符为"\",而在Unix/Linux系统中,路径分隔符为"/"。因此,在使用os模块拼接路径时,最好使用os.path.join()函数来确保路径正确。
阅读全文