pandasのDataFrameで特定の列に書かれた数値を、上から○段目までかを選択してから四則演算で算出したい

Question

前提・実現したいこと

調べても方法が見当たらなかったのでこちらへ質問をさせていただきます。

pandasのDataframeで読み込んだCSVの表にC1,C2,C3という列が存在していると仮定して、その中からC2に載っている数値の『上から２番目』までの数値のみを選択して四則演算で結果を表示したい場合はどのような関数を入力すればよいのでしょうか？

拙い文章で申し訳ありませんが、ご回答頂けましたら幸いです。

（ソースコードにCSV内の表のイメージを記載しておきます）

該当のソースコード

#     C1   C2   C3
# 0   24   24   31
# 1   18   38   24
# 2   19   11   15

Comments

[TakaiY]

「C2に載っている数値の『上から２番目』までの数値のみを選択して四則演算」をしたいということですが、具体的な例を示してください。
例のデータであれば、"24 + 38" のような計算をやりたいということですか？

[mamegorou0083]

はい、おっしゃる通りでございます。

Answer 1

C2に載っている数値の『上から２番目』までの数値のみを選択して四則演算で結果を表示したい

質問に文字通り答えるとこうなります。

>>> print(df)
   C1  C2  C3
0  24  24  31
1  18  38  24
2  19  11  15
>>> print(df['C2'][0] + df['C2'][1])
62
>>> print(df['C2'][0] - df['C2'][1])
-14
>>> print(df['C2'][0] * df['C2'][1])
912
>>> print(df['C2'][0] / df['C2'][1])
0.631578947368421
>>>

タイトルが「上から○段目」なのですね。
足し算なら以下です。

>>> print(df)
   C1  C2  C3
0  24  24  31
1  18  38  24
2  19  11  15
3  35  18  21
>>>
>>> def sum_cn(df, column, n):
...     return sum(df[column].values[:n])
...
>>> print(sum_cn(df, 'C2', 2))
62
>>> print(sum_cn(df, 'C2', 3))
73
>>> print(sum_cn(df, 'C2', 4))
91
>>>

掛け算なら以下です。

>>> import functools
>>> import operator
>>>
>>> print(df)
   C1  C2  C3
0  24  24  31
1  18  38  24
2  19  11  15
3  35  18  21
>>>
>>> def pi_cn(df, column, n):
...     return functools.reduce(operator.mul, df[column].values[:n], 1)
...
>>> print(pi_cn(df, 'C2', 2))
912
>>> print(pi_cn(df, 'C2', 3))
10032
>>> print(pi_cn(df, 'C2', 4))
180576

Comments

[mamegorou0083]

上記のコードを試してみたところ、正しく処理が行えました！
ご教示くださり本当にありがとうございました。