熊猫dataframe中的F弦格式

Question

假设“成员.xlsx”包括与成员的数据相对应的列，例如“ country”，“ name”，“ job”，“ age”，“ hobby”，“ hepth”，“ height”，“ stright”等。我能够要弄清楚如何提取成员的“名称”列，其县是美国，如下面使用熊猫。

import pandas as pd
df=pd.read_excel('f:/temporary/members.xlsx') 
df1=df.loc[df['Country']=='US', ['Name']]
df1.to_excel('f:/temporary/US_Names.xlsx')

现在，我还想提取其他国家成员（例如，“希腊”，“印度”）的其他列（例如，“工作”，“年龄”和“爱好”）。我想的是使用F-string替换“我们”和“名称”，这与此相似（我知道'df1 ='也应该更改，但稍后让我处理...）。

for a in ['US','Greece','India']:
    for b in ['Name','Job','Age','Hobby']:
        df1=df.loc.[df['Country']=={a}, [{b}]]

但是在'df1 = df.loc中。[df ['country'] == {a}，[{b}]]'，我不知道应该在f-string格式化的位置放置'f'。即使不是F弦格式，任何方法都将起作用。提前致谢。

环境：Windows 10 64位，Python 3.10.4，Pandas 1.4.1

Answer 1

使用：

for a in ['US','Greece','India']:
    for b in ['Name','Job','Age','Hobby']:
        df1=df.loc[df['Country']==a, b]

如果您不需要分开具有不同价值的国家的DF，请使用：

for a in ['US','Greece','India']:
   df1=df.loc[df['Country']==a, ['Name','Job','Age','Hobby']]

演示：

df=pd.DataFrame({'Name':['ali', 'reza'], 'Country':['Ir', 'Gr']})
a='Ir'
b='Name'
df.loc[df['Country']==a, b]

输出：

0    ali
Name: Name, dtype: object