pandas中curr函数报错ValueError: could not convert string to float: ‘RL‘解决办法
报错情况(ValueError)
# 代码块
corrmat = df_train.corr()
plt.figure(figsize=(16, 8),dpi=600)
seaborn.heatmap(corrmat, square=True, cmap='YlGnBu', xticklabels = True , yticklabels = True)
plt.show()
ValueError: could not convert string to float: ‘RL’
原因分析
curr()
函数原本可以自动忽略字符串,但pandas2.0版本后,原来corr()
函数自动忽略 字符串 等非浮点数的特性被修改。 现在需要加上numeric_only=True
才会忽略字符串。
修改代码
# 代码块
corrmat = df_train.corr(numeric_only=True)
plt.figure(figsize=(16, 8),dpi=600)
seaborn.heatmap(corrmat, square=True, cmap='YlGnBu', xticklabels = True , yticklabels = True)
plt.show()