基于列值的熊猫指数设置

Question

我是熊猫新手，我有一个从excel文件中读取的Dataframe，比如下面的屏幕截图，其中Products是标题，Google和Meta是组(索引)。

df = pd.DataFrame({
    "products": ["Google", "Youtube", "Fitbit", "Nest", "Waze", "Meta", "Facebook", "Instagram", "Whatsapp"]
})

df

    products
0   Google
1   Youtube
2   Fitbit
3   Nest
4   Waze
5   Meta
6   Facebook
7   Instagram
8   Whatsapp

​

​

使用熊猫，我希望数据采集器是这样的

​

​

谢谢

Answer 1

使用pd.assign和np.where

import numpy as np
import pandas as pd


companies = ["Google", "Meta"]
df = df.assign(
    companies=np.where(df["products"].isin(companies), df["products"], "")
).set_index("companies")

df["products"] = df["products"].replace(companies, "")
print(df)

            products
companies           
Google              
             Youtube
              Fitbit
                Nest
                Waze
Meta                
            Facebook
           Instagram
            Whatsapp