熊貓選擇多列數據的最佳做法

Question

我目前有一個非常冗餘的代碼來處理來自熊貓數據框的一行中的多個列。這看起來是這樣的：

info = [] 
info.append(row['column name']) 
... 
(multiple lines of mere appending) 

而且我傳遞到填充在網頁上返回的值另一個Python腳本這個名單。我必須改進這一點以擺脫冗餘，我不知道如何去解決這個問題。

做這種工作的最佳做​​法是什麼？

Answer 1

您可以傳遞列名稱列表。

import pandas as pd 

df = pd.DataFrame({'x':[1, 2, 3], 'y':[5, 6, 7], 'z':['a', 'b', 'c']}) 

選擇一排iloc和使用名稱的列表中選擇列，例如如果你只是想 'X' 和 'Z'：

df.iloc[1][['x', 'z']] 

回報：

x 2 
z b 
Name: 1, dtype: object