python 读取文件夹内的csv文件,并且记录文件名称,分别取第四列数据作为y变量,第七列数据作为x变量,作图 代码
时间: 2023-06-06 22:07:46 浏览: 20
import os
import csv
import matplotlib.pyplot as plt
folder_path = "path/to/folder"
x_data = []
y_data = []
file_names = []
for filename in os.listdir(folder_path):
if filename.endswith(".csv"):
file_names.append(filename)
with open(os.path.join(folder_path, filename), 'r') as csv_file:
csv_reader = csv.reader(csv_file)
next(csv_reader) #skip header row
for row in csv_reader:
x_data.append(float(row[6]))
y_data.append(float(row[3]))
plt.scatter(x_data, y_data)
plt.title("Scatter Plot of x vs y")
plt.xlabel("x")
plt.ylabel("y")
plt.show()
# file_names list can be used to print the names of the files used for the plot