擅长:python、mysql、java
<p>您可以创建一个自定义布尔值来<code>groupby</code>,并在字典中创建两个单独的数据帧。假设<code>Inv_Type</code>中只有两个值,那么我们可以使用<code>nunique</code>来fidn任何具有多个值的组</p>
<pre><code>dfs = {int(grp) : data for grp,data
in df.groupby([df.groupby('Product')['Inv_Type'].transform('nunique') > 1])}
print(dfs[1])
Product Inv_Type Quant
12 Prod3 X 2
13 Prod3 Y 12
14 Prod3 Y 2
15 Prod3 X 19
16 Prod3 Y 6
17 Prod3 X 5
18 Prod3 X 4
19 Prod3 Y 13
print(dfs[0])
Product Inv_Type Quant
0 Prod1 X 16
1 Prod1 X 13
2 Prod1 X 8
3 Prod1 X 16
4 Prod2 Y 14
5 Prod2 Y 10
6 Prod2 Y 4
7 Prod2 Y 13
8 Prod2 Y 7
9 Prod2 Y 16
10 Prod2 Y 13
11 Prod2 Y 11
20 Prod5 X 11
21 Prod5 X 10
22 Prod5 X 13
23 Prod5 X 10
</code></pre>