當 column_name 是 Python Pandas 中的變量時，獲取 df.loc['column_name']

Question

我有一個變量，我稱之為“響應”，用戶手動輸入“Apple”或“Tree”這些詞。 然后我想根據用戶回答 select 名稱=“Apple”或“Tree”的列。 我怎樣才能做到這一點？ 像 df.loc[response] 這樣的東西？

任何幫助將不勝感激，謝謝

Answer 1

您可以簡單地訪問具有response變量的列，例如

df[response]

df[response]將根據您在響應變量中的內容評估為df['Apple']或df['tree']

Answer 2

最簡單的方法是：

df[response]

其中 response 是變量，可以是“Apple”或“Tree”。