温馨提示×

SQL在DataFrame中的子查询如何实现

sql
小樊
83
2024-09-09 22:35:33
栏目: 云计算

在Python的Pandas库中,DataFrame对象支持类似于SQL的子查询操作

首先,我们需要创建一个示例DataFrame。这里我们使用Pandas创建一个简单的数据集:

import pandas as pd

data = {'A': [1, 2, 3, 4],
        'B': [5, 6, 7, 8],
        'C': [9, 10, 11, 12]}

df = pd.DataFrame(data)
print(df)

假设我们想要找到列’B’中大于5的所有行,并将结果存储在一个新的DataFrame中。我们可以使用以下代码实现这个功能:

sub_query_result = df[df['B'] > 5]
print(sub_query_result)

这将输出以下结果:

   A  B   C
1  2  6  10
2  3  7  11
3  4  8  12

这就是在Pandas DataFrame中实现子查询的方法。你可以根据自己的需求修改条件和筛选的列。

0