按组查找数量 _编程开发

创始人

2024-09-02 20:00:21

0次

假设我们有一个数据集，其中包含一些组和它们的数量。

首先，我们需要导入pandas库并读取我们的数据集。

import pandas as pd

# 读取数据集
dataset = pd.read_csv('dataset.csv')

# 查看数据集
print(dataset.head())

接下来，我们可以使用groupby()函数按组对数据集进行分组，并取出每个组的数量（即数量列）。

# 按组分组并计算数量
grouped_data = dataset.groupby('组')['数量'].sum()

# 显示结果
print(grouped_data)

最后，我们可以将结果放入一个DataFrame对象中，以便更容易地查看结果。

# 将结果放入数据框中
result_df = pd.DataFrame(grouped_data)

# 显示最终结果
print(result_df)

在这个例子中，我们假设数据集中有以下内容：

使用上述代码，我们可以得到以下结果：

组
A    60
B    75
Name: 数量, dtype: int64

   数量
组    
A  60
B  75

这说明我们成功地按组查找了数量，并将结果放入了一个数据框中，以便更容易地查看。