python - Pandas sql 等效项

Question

我有一个包含以下字段的 pd.dataframe:id, value(每个 id 有多个值)。

pandas 相当于 sql 查询:

SELECT id, Max(value)-Min(value) AS val1
FROM t1
GROUP BY t1.id

Answer 1

你可以这样做:

In [31]: df = pd.DataFrame(np.random.randint(0, 5, (10, 2)),  columns=['id','value'])

In [32]: df
Out[32]:
   id  value
0   2      4
1   4      0
2   3      1
3   4      2
4   4      1
5   2      3
6   1      0
7   3      2
8   2      2
9   1      1

In [33]: df.groupby('id')['value'].apply(lambda x: x.max() - x.min()).reset_index()
Out[33]:
   id  value
0   1      1
1   2      2
2   3      1
3   4      2

这里是Pandas comparison with SQL有很多例子 - 这可能有用