如何在python中使用loc选择和显示多列

Question

假设我有一个数据帧（ds），如下所示：

ID Name Age                
1  xxc 34               
2  sfg 23    
3  hdg 18

我也想显示列Name和Age。

当前通过这一行代码

def item(id):
   return ds.loc[ds['ID'] == id]['Name'].tolist()[0]

我只能获取Name列值。我如何也获得Age列值？

请注意，我想保留相同的代码，即return语句。 有什么解决办法吗？

Answer 1

list(ds.loc[ds['ID'] == 1,['Name','Age']].iloc[0])
Out[365]: ['xxc', 34]