熊猫corr()经常返回NaN

时间:2018-09-23 13:59:59

标签: python pandas dataframe statistics correlation

我试图在数据帧上运行我认为应该是简单的相关函数的函数,但是它在我认为不应该的地方返回NaN。

代码:

# setup
import pandas as pd
import io

csv = io.StringIO(u'''
id  date    num
A   2018-08-01  99
A   2018-08-02  50
A   2018-08-03  100
A   2018-08-04  100
A   2018-08-05  100
B   2018-07-31  500
B   2018-08-01  100
B   2018-08-02  100
B   2018-08-03  0
B   2018-08-05  100
B   2018-08-06  500
B   2018-08-07  500
B   2018-08-08  100
C   2018-08-01  100
C   2018-08-02  50
C   2018-08-03  100
C   2018-08-06  300
''')

df = pd.read_csv(csv, sep = '\t')

# Format manipulation
df = df[df['num'] > 50]
df = df.pivot(index = 'date', columns = 'id', values = 'num')
df = pd.DataFrame(df.to_records())

# Main correlation calculations
print df.iloc[:, 1:].corr()

主题DataFrame:

       A      B      C
0    NaN  500.0    NaN
1   99.0  100.0  100.0
2    NaN  100.0    NaN
3  100.0    NaN  100.0
4  100.0    NaN    NaN
5  100.0  100.0    NaN
6    NaN  500.0  300.0
7    NaN  500.0    NaN
8    NaN  100.0    NaN

corr()结果:

    A    B    C
A  1.0  NaN  NaN
B  NaN  1.0  1.0
C  NaN  1.0  1.0

根据函数上的(有限)documentation,它应排除“ NA /空值”。由于每一列都有重叠的值,因此结果是否应全部不是非NaN?

讨论herehere很好,但都没有回答我的问题。我已经尝试过讨论herefloat64想法,但这也失败了。

@hellpanderr的评论提出了一个很好的观点,我使用的是0.22.0

奖金问题-我不是数学家,但是在这个结果中B和C之间如何存在1:1的相关性?

1 个答案:

答案 0 :(得分:5)

结果似乎是您处理的数据的伪影。在您撰写本文时,NA被忽略,因此基本上可以归结为:

df[['B', 'C']].dropna()

       B      C
1  100.0  100.0
6  500.0  300.0

因此,每列只剩下两个值用于计算,因此应该lead to to correlation coefficients of 1

df[['B', 'C']].dropna().corr()

     B    C
B  1.0  1.0
C  1.0  1.0

那么,NA的其余组合从何而来?

df[['A', 'B']].dropna()

       A      B
1   99.0  100.0
5  100.0  100.0


df[['A', 'C']].dropna()

       A      C
1   99.0  100.0
3  100.0  100.0

因此,同样在这里,您最终每列只有两个值。区别在于,列BC仅包含一个值(100),该值的标准偏差为0

df[['A', 'C']].dropna().std()

A    0.707107
C    0.000000

计算相关系数时,用标准偏差除以NA