读取excel数据,使用xlrd制作散点图。
import matplotlib.pyplot as plt
import xlrd
data = xlrd.open_workbook('测试3.xlsx') # 读取工作表
table = data.sheet_by_name('Sheet1') # 读取当前sheet表对象
rows = table.nrows # 获取行数
nianlist = []
citylist = []
salarylist = []
print('一共有{}行数据'.format(rows))
for i in range(1,rows):
salary = table.row_values(i)[3]
nian = table.row_values(i)[6]
city = table.row_values(i)[8]
nianlist.append(nian)
citylist.append(city)
salarylist.append(salary)
plt.scatter(citylist, salarylist, s=10)
plt.show()