代码之家 › 专栏 › 技术社区 › dzieciou

多栏高效搜索

performance pandas python

dzieciou · 技术社区 · 4 年前

我有 DataFrame

df = pd.DataFrame([
    {
        'name': 'Alexandra',
        'surname': 'Johnson',
        'age': 5
    },
    {
        'name': 'Roland',
        'surname': 'Hawking',
        'age': 3
    }
])

df[((df['name']=='Roland') | (df['surname']=='Johnson')) & (df['age'] > 4)]

        name  surname  age
0  Alexandra  Johnson    5

indexed = df.set_index('name')
indexed[indexed.index == 'Roland']

对于多个列,我可以使用 MultiIndex

1 回复 | 直到 4 年前

sammywemmy 4 年前

我不确定你指的是不是这个,所以我要试试看:

将名称和姓氏设置为索引

df = df.set_index(['name','surname'])

df.loc[('Alexandra', 'Johnson')]

    age    5
Name: (Alexandra, Johnson), dtype: int64

推荐文章

July · 如何定义数字间隔,然后四舍五入

1 年前

Community wiki · 对象名称前的单下划线和双下划线的含义是什么?

1 年前

Brian Johnson · 为什么在Python中列出字典列表会引发TypeError?[已关闭]

1 年前

user026 · 如何根据特定窗口的平均值(行数)创建新列?

1 年前

Ashok Shrestha · 需要追踪特定的颜色线并获取坐标

1 年前

Nicote Ool · 在FastApi和Vue3中获得422

1 年前

NeoExceptCodeBad · 如果我有很多垂直线,我如何找到它们的边缘?

1 年前

Abdulaziz · 如何对集合内的列表进行排序[重复]

1 年前

user2743931 · 带有src目录的Python setup.py

1 年前

asmgx · 为什么合并数据帧不能按照python中的预期方式工作

1 年前