python 读取文件夹内的csv文件,并且记录文件名称,分别取第四列数据作为y变量,第七列数据作为x变量,作图 代码
时间: 2023-06-06 12:07:46 浏览: 86
moxing_python数据清洗_读取csv文件_数据清洗_
5星 · 资源好评率100%
下面是Python代码的实现,可以读取文件夹内的CSV文件,并记录文件名、取指定列作为变量并作图:
```python
import os
import matplotlib.pyplot as plt
import pandas as pd
# 定义CSV数据列数
Y_COLUMN_INDEX = 3
X_COLUMN_INDEX = 6
# 定义CSV文件所在文件夹路径
FOLDER_PATH = "path/to/folder"
# 定义存储文件名、X、Y数据的列表
files = []
x_data = []
y_data = []
# 循环处理CSV文件
for filename in os.listdir(FOLDER_PATH):
if filename.endswith(".csv"):
# 读取CSV文件数据
df = pd.read_csv(os.path.join(FOLDER_PATH, filename))
# 记录文件名
files.append(filename)
# 取第四列数据作为Y变量,第七列数据作为X变量
y = df.iloc[:, Y_COLUMN_INDEX]
x = df.iloc[:, X_COLUMN_INDEX]
# 记录X、Y数据
x_data.append(x)
y_data.append(y)
# 绘制图形
plt.plot(x, y, label=filename)
# 定义图例和坐标轴信息
plt.legend()
plt.xlabel("X")
plt.ylabel("Y")
# 显示图形
plt.show()
```
请注意,此代码并不完美或最佳,每个任务都有不同的解决方案,可以使用其他方法实现此任务。
阅读全文