擅长:python、mysql、java
<p>另一种方法是将自定义方法传递给<code>pandas.DataFrame.corr()</code>函数</p>
<pre><code>from sklearn.feature_selection import mutual_info_regression
def custom_mi_reg(a, b):
a = a.reshape(-1, 1)
b = b.reshape(-1, 1)
return mutual_info_regression(a, b)[0] # should return a float value
df_mi = df.corr(method=custom_mi_reg)
</code></pre>