python 读取文件夹内的csv文件,并且记录文件名称,分别取第四列数据作为y变量,第七列数据作为x变量,作图 代码
时间: 2023-06-06 13:07:41 浏览: 138
以下是Python代码实现:
```python
import os
import csv
import matplotlib.pyplot as plt
# 获取文件夹路径
dir_path = 'your_directory_path'
# 存储所有读取的csv文件数据
data = {}
# 遍历文件夹中的csv文件
for filename in os.listdir(dir_path):
if filename.endswith('.csv'):
file_path = os.path.join(dir_path, filename)
with open(file_path, 'r') as f:
reader = csv.reader(f)
# 跳过csv文件的首行
next(reader)
# 读取csv文件中的数据
file_data = []
for row in reader:
x = float(row[6]) # 获取第七列数据作为x变量
y = float(row[3]) # 获取第四列数据作为y变量
file_data.append((x, y))
# 将数据存储到字典中
data[filename] = file_data
# 根据读取的数据绘制图形
for filename, file_data in data.items():
x = [i[0] for i in file_data]
y = [i[1] for i in file_data]
# 绘制散点图
plt.scatter(x, y, label=filename)
plt.xlabel('x')
plt.ylabel('y')
plt.legend()
plt.show()
```
请注意,以上代码只是一个示例,具体实现需要根据csv文件的格式作出相应修改。
阅读全文