import numpy as np import scipy.stats as ss import pandas as pd import matplotlib.pyplot as plt import seaborn as sns df=pd.read_csv("./HR.csv") dp_indices=df.groupby(by="department").indices sales_values=df["left"].iloc[dp_indices["sales"]].values technical_vlues=df["left"].iloc[dp_indices["technical"]].values dp_keys=list(dp.indices.keys()) dp_t_mat=np.zeros([len(dp_keys)],[len(dp_keys)]) for i in range(len(dp_keys)): for j in range(len(dp_keys)): p_vlaue=ss.ttest_ind(df["left"].iloc[dp_indices[dp_keys[i]]].values),\ df["left"].iloc[dp_indices[dp_keys[j]]].values[1] dp_t_mat[i][j]=p_value sns.heatmap(dp_t_mat,xticklabels=dp_keys,yticklabels=dp_keys) plt.show()提示name 'dp' is not defined
时间: 2023-11-27 17:50:09 浏览: 50
看起来你的代码中有一个变量名错误。在第4行,你将 `df.groupby(by="department").indices` 存储在变量 `dp_indices` 中,而在第6行,你试图使用 `dp` 来引用 `dp_indices`。因此,您需要将第6行中的 `dp` 替换为 `dp_indices`,以便代码可以正常工作。下面是修改后的代码:
```
import numpy as np
import scipy.stats as ss
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
df=pd.read_csv("./HR.csv")
dp_indices=df.groupby(by="department").indices
sales_values=df["left"].iloc[dp_indices["sales"]].values
technical_values=df["left"].iloc[dp_indices["technical"]].values
dp_keys=list(dp_indices.keys())
dp_t_mat=np.zeros([len(dp_keys),len(dp_keys)])
for i in range(len(dp_keys)):
for j in range(len(dp_keys)):
p_value=ss.ttest_ind(df["left"].iloc[dp_indices[dp_keys[i]]].values,
df["left"].iloc[dp_indices[dp_keys[j]]].values)[1]
dp_t_mat[i][j]=p_value
sns.heatmap(dp_t_mat,xticklabels=dp_keys,yticklabels=dp_keys)
plt.show()
```
请注意,我还更正了第10行和第12行中的变量名拼写错误。