Question

这有点难以解释所以请耐心等待。

假设我有一张表，如下所示

如何创建符合以下条件的新数据框

每行有5行，它们是A列中一个范围之间的值，例如第一行介于（200,311）之间，第二行介于（312,370）之间等。
每列有3列，将是B列中一个范围之间的值，例如第一列介于（1,16）之间，第二列介于（17,50之间）等。
每个单元格的值将是C列中与相应列和行匹配的值的总和。

示例：

任何插图？数字是随机的，你不需要按照我的例子。

非常感谢！

我的解决方案是在两个列表中预定义行标准和列标准，然后运行嵌入式循环以将每个单元格值填充到新数据帧中。它工作而不是那么慢，但我想知道因为这是pandas数据帧，在查询中应该有这样做，没有任何循环。

再次感谢！

Answer 1

您可以使用cut获取范围，然后将其提供给pivot_table以获取总和：

# Setup example data.
np.random.seed([3, 1415])
n = 100
df = pd.DataFrame({
    'A': np.random.randint(200, 601, size=n),
    'B': np.random.randint(1, 101, size=n),
    'C': np.random.randint(25, size=n)
    })

# Use cut to get the ranges.
a_bins = pd.cut(df['A'], bins=[200, 311, 370, 450, 550, 600], include_lowest=True)
b_bins = pd.cut(df['B'], bins=[1, 16, 67, 100], include_lowest=True)

# Pivot to get the sums.
df2 = df.pivot_table(index=a_bins, columns=b_bins, values='C', aggfunc='sum', fill_value=0)

结果输出：

B           [1, 16]  (16, 67]  (67, 100]
A                                       
[200, 311]       82       118        153
(311, 370]       68        56         45
(370, 450]       41       129         40
(450, 550]       32       121         57
(550, 600]        0       112         47

Answer 2

我真的很喜欢@root's solution！这是一个略微修改的单行版本，它使用pd.crosstab方法：

In [102]: pd.crosstab(
     ...:     pd.cut(df['A'], bins=[200, 311, 370, 450, 550, 600], include_lowest=True),
     ...:     pd.cut(df['B'], bins=[1, 16, 67, 100], include_lowest=True),
     ...:     df['C'],
     ...:     aggfunc='sum'
     ...: )
     ...:
Out[102]:
B           [1, 16]  (16, 67]  (67, 100]
A
[200, 311]       31       157        117
(311, 370]       23        90         38
(370, 450]      110       168         60
(450, 550]       37       117        115
(550, 600]       35        19         49

通过对多个列进行分组来过滤python pandas数据帧

2 个答案: