numpy工作数据统计整理

需求：
（1）员工的平均薪资为多少？
（2）公司任职最久的员工是谁？
（3）员工的平均工作年限是多少？
（4）员工总体流失率是多少？ # 离职人数/总人数
（5）员工整体满意程度如何？ # 平均满意度

import numpy as np
lol = np.load("lol_data.npz")
data = lol["data"]  # 获取数据
columns = lol["columns"]  # 获取表格属性
# print(data, columns)
print("员工的平均薪资为：", np.average(data[:, 4].astype(int)))
time = data[:, -3].astype(int)
long_time_index = np.argmax(time)
print("公司任职最久的员工是：", data[long_time_index, 1])
print("员工的平均工作年限是：", np.average(data[:, -3].astype(int)))
leave = [i == "离职" for i in data[:, -1]]
print("员工总体流失率为：", len(data[leave])/len(data))
print("员工总体满意度为：", np.average(data[:, -2].astype(int)))