import csv
import cv2 as cv
import numpy as np
filePath = 'F:/zhengwangwork/test csv/11.csv'
# filePath = input("请输入处理文件存放的位置：")
def readcsv():
# file = open("D:/Work/csv/31.csv")
# filename=input("请输入要读取文件路径：")
file = open("F:/csvFile/2.csv")
# file = open(filename)
reader = csv.reader(file)
data = []
data1 = []
data2 = []
data3 = []
data4 = []
data5 = []
a = []
# 读取文件数据列
for row in reader:
# column = eval(row[0])
column = row[0]
# min=min(column)
column1 = row[1]
column2 = row[2]
column3 = row[3]
column4 = row[4]
column5 = row[5]
data.append(column)
data1.append(column1)
data2.append(column2)
data3.append(column3)
data4.append(column4)
data5.append(column5)
# 求取第一列的最小值并相减输出
q = min(data)
data = [eval(i) - eval(q) for i in data]
# print(data)
# 将数据列重新写入csv文件中
# filePath = 'F:/zhengwangwork/test csv/6.csv'
rows = zip(data, data1, data2, data3, data4, data5)
with open(filePath, "w", newline='') as f:
writer = csv.writer(f)
for row in rows:
writer.writerow(row)
def drawpicture():
# (233,1600,3)分别是宽、高、通道数
canvas = np.zeros((233, 1600, 3), dtype=np.uint8) + 255
file = open(filePath) # 打开csv文件
reader = csv.reader(file) # 读取csv文件
data = list(reader) # 将csv数据转化为列表
length_h = len(data) # 得到数据行数
lenght_l = len(data[0]) # 得到每行长度
x = list()
y = list()
z = list()
# 绘制散点图
for i in range(0, length_h): # 从第一行开始读取
x.append(int(eval(data[i][0]))) # 将第一列数据从第一行读取到最后一行付给列表x
y.append(eval(data[i][2])) # 将第三列数据从第一行读取到最后一行付给列表y
z.append(eval(data[i][1]))
pts = zip(x, y)
col = []
dict = {"0":(0,255,255), "1": (0,0,255), "2": (255,192,255),"3": (255,127,255), "4": (255,64,255),
"5": (127,255,0),
"6": (127,0,255), "7": (64,0,255), "8": (0,255,0), "9": (0,127,0), "10": (255,192,0),
"11": (255,127,0), "12": (255,64,0), "13": (127,55,255), "14": (127,0,0), "15": (64,0,0),
"16": (0,0,127), "17": (255,0,255), "18":( 255,0,0)," 19": (127,240,0), "20": (127,240,255),
"21": (255,15,255), "22": (255,15,0), "23": (0,0,120), "24": (255,255,255), "25": (255,255,0),
" 26": (255,240,255), "27": (255,240,0)}
# print(dict["0"])
# # 通过循环将不同的通道值对应的颜色加到col数组中，其中data1就是db数组
for i in range(0, len(z)):
col.append(dict[str(z[i])])
i = 0
# 画出每个点，颜色随机
for pt in pts:
print(pt)
cv.circle(canvas, pt, 2, col[i],-1)
i = i + 1
# 展示绘制的图形
cv.namedWindow('Basic Graph', cv.WINDOW_NORMAL)
cv.imshow('Basic Graph', canvas)
# 保存绘制的图像
cv.imwrite('Basic_Graph.png', canvas)
cv.waitKey()
if __name__ == '__main__':
readcsv()
drawpicture()
