随机交换两列熊猫数据框的元素

Question

如何以给定的概率随机交换两列pandas DataFrame的元素？这样做需要很长时间：

import random

for i in range(len(df)):
 if random.random() < 0.5:
  df.loc[i, 'A'], df.loc[i, 'B'] = df['B'].loc[i], df['A'].loc[i]

df

   A   B
0  9   14      
1  1   32     
2  8   23 
3  1    2    
4  10   66 

有一个类似的问题here

Answer 1

试试：

# sample data
df = pd.DataFrame(np.arange(20).reshape(10,2), columns=['A','B'])

# random indexing, seed for repeatability
# remove seed for randomness
np.random.seed(42)
idx = np.random.rand(len(df)) < 0.5

# passing numpy array to bypass column alignment
df.loc[idx, ['A','B']] = df.loc[idx, ['B','A']].to_numpy()

输出：

    A   B
0   1   0
1   2   3
2   4   5
3   6   7
4   9   8
5  11  10
6  13  12
7  14  15
8  16  17
9  18  19