你是否在寻找 pandas
import pandas as pd
# 原始数据
a = [
{"id": 214, "company": "CN920A"},
{"id": 225, "company": "CN920A"},
{"id": 239, "company": "CN920A"},
{"id": 226, "company": "CN920E"},
{"id": 236, "company": "CN920E"},
{"id": 194, "company": "CN920E"},
{"id": 193, "company": "CN930D"},
{"id": 192, "company": "CN930D"},
{"id": 191, "company": "CN940F"},
{"id": 190, "company": "CN940F"}
]
# 将数据转换为 DataFrame
df = pd.DataFrame(a)
# 按 company 进行分组,并将相同 company 的 id 放到一个列表中
result = df.groupby('company')['id'].apply(list).reset_index()
# 输出最终结果
print(result)
company id
0 CN920A [214, 225, 239]
1 CN920E [226, 236, 194]
2 CN930D [193, 192]
3 CN940F [191, 190]
|