熊猫根据行值（条件）创建新列

Question

我有一个这样的专栏

A
1.0
1.0
2.0
3.0
4.0
5.0
5.0
5.0

我需要根据条件创建一个新列，如果 a[i] 和 a[i-1] 相同，则值为 0，否则为 1。

结果应该是这样的：

A       B
1.0     1
1.0     0
2.0     1
3.0     1   
4.0     1
5.0     1   
5.0     0   
5.0     0

正确的熊猫方式来做到这一点？

Answer 1

通过 sompare 为不等于ne与shift ed Series创建布尔掩码并shift为integer ：

df['B'] = df['A'].ne(df['A'].shift()).astype(int)
print (df)
     A  B
0  1.0  1
1  1.0  0
2  2.0  1
3  3.0  1
4  4.0  1
5  5.0  1
6  5.0  0
7  5.0  0

详情：

print (df['A'].ne(df['A'].shift()))
0     True
1    False
2     True
3     True
4     True
5     True
6    False
7    False
Name: A, dtype: bool