如何編寫 Python 代碼來查找特定行值的 Pandas DF 中列的值的總和？

Question

包含幾列的 Pandas 數據框。 讓我們說年，產量，單位數量，每單位價格。

我知道如何使用 sum() 函數獲得 DF 中所有現有年份的總產量。

現在我想找到特定年份的產量，比如說 2014 年。現在這些年份在數據框中重復。 當 year = 2014 時，如何編寫代碼來查找數據框中所有生產值的總和？

我是否使用 groupby 功能？ 但是我如何有效地做到這一點以達到這個結果？

謝謝！

Answer 1

對您列出的變量使用任意值，

import pandas as pd

years = list(range(2010, 2020))
years.extend([2014, 2014])
production = list(range(12))
num = list(range(12))
price = list(range(12))

df = pd.DataFrame({'years':years, 'prod': production, 'num': num,\
        'price': price})
answer = df[df['years'] == 2014]['prod'].sum().item()

print(answer)