Question

以下代码将产生一个数据帧，该数据帧代表四个玩家，每个玩家掷出两个掷骰子的骰子：

mux = pd.MultiIndex.from_arrays([
    list(['Alice', 'Alice', 'Alice', 
          'Bob', 'Bob', 'Bob', 'Bob', 'Bob', 
          'Carol', 'Carol', 'Carol', 'Carol', 'Carol', 'Carol', 
          'Dan', 'Dan']),
    list('1231234512345612')
], names=['player', 'roll'])

#df = pd.DataFrame({'die1': np.random.randint(1, 7, len(mux)),
#                   'die2': np.random.randint(1, 7, len(mux))}, mux)

df = pd.DataFrame({'die1': ['3','6','2','1','1','6','5','1','3','1','4','5','3','5','5','5'],
                   'die2': ['2','5','1','4','1','6','3','5','6','4','5','1','4','4','3','6']}, mux)

这是产生的数据框：

            die1 die2
player roll          
Alice  1       3    2
       2       6    5
       3       2    1
Bob    1       1    4
       2       1    1
       3       6    6
       4       5    3
       5       1    5
Carol  1       3    6
       2       1    4
       3       4    5
       4       5    1
       5       3    4
       6       5    4
Dan    1       5    3
       2       5    6

我对玩家感兴趣-不仅是个人掷骰，而且是玩家-在第一个骰子上掷出6。

以下代码执行了我不想要的操作，它返回第一个骰子为6的单个骰子：df[df.die1=='6']：

             die1   die2
player  roll        
Alice   2    6      5
Bob     3    6      6

我如何通过使用die1产生以下内容，而没有提及各个玩家的姓名，我显然不知道先验？

            die1 die2
player roll          
Alice  1       3    2
       2       6    5
       3       2    1
Bob    1       1    4
       2       1    1
       3       6    6
       4       5    3
       5       1    5

Answer 1

您可以执行groupby：

# is your data string '6' or integer 6
df[df['die1'].eq('6').groupby('player').transform('any')]

输出：

            die1 die2
player roll          
Alice  1       3    2
       2       6    5
       3       2    1
Bob    1       1    4
       2       1    1
       3       6    6
       4       5    3
       5       1    5

Answer 2

ll=[]
for i, j in df.groupby('player'):
    if len(j[j.die1==6])>0:
        ll.append(j)
return pd.concat(ll)

熊猫：多索引子集选择

2 个答案: