pandasの関数内での計算

Question

pandasをdefで定義した関数内で計算し、その結果を扱いたいです。

いつも大変お世話になっています。
簡単なコードなのですが、pandasのことについてご教授頂ければと思います。

該当のソースコード

def func(df):
    af=pd.DataFrame(np.arange(3*3).reshape(3,3))
    df=df.add(af, fill_value=0.0)

global df
df=pd.DataFrame(np.arange(3*3).reshape(3,3))
func(df)
df

結果

    0   1   2
0   0   1   2
1   3   4   5
2   6   7   8

期待する結果
    0   1   2
0   0   2   4
1   6   8  10
2  12  14  16

試したこと

関数内を、

def func(df):
    af=pd.DataFrame(np.arange(3*3).reshape(3,3))
    return df.add(af, fill_value=0.0)

にしてみましたが駄目でした。
すみませんが、お知恵をお貸し頂けると嬉しいです。

Answer 1

dfのshapeが不定なら

import numpy as np
from operator import mul
from functools import reduce

def func(df):
    cnt = reduce(mul, df.shape)
    af = pd.DataFrame(np.arange(cnt).reshape(df.shape))
    df = df.add(af)
    return df

df = pd.DataFrame(np.arange(3*3).reshape(3,3))
df = func(df)

そもそも単純に全要素を２倍したいということなら

import numpy as np
df = pd.DataFrame(np.arange(3*3).reshape(3,3))
df = df.applymap(lambda e:e*2)

Comments

[ryooope]

連絡が遅れてすみません。
無事できました！

Answer 2

def func(df):
    af=pd.DataFrame(np.arange(3*3).reshape(3,3))
    df=df.add(af, fill_value=0.0)
    return df

global df
df=pd.DataFrame(np.arange(3*3).reshape(3,3))
df=func(df)
df

【追記】

import pandas as pd
import numpy as np

def func(df):
    af=df.copy()
    df=df.add(af, fill_value=0.0)
    return df

global df

df=pd.DataFrame(np.arange(4*4).reshape(4,4))
print(df)
#    0   1   2   3
#0   0   1   2   3
#1   4   5   6   7
#2   8   9  10  11
#3  12  13  14  15

df=func(df)
print(df)
#    0   1   2   3
#0   0   2   4   6
#1   8  10  12  14
#2  16  18  20  22
#3  24  26  28  30

Comments

[ryooope]

解答ありがとうございます。
3*3ならできたのですが、4*4, 5*5等をするとやはり同じ結果になるのですが何かご存知でしょうか？

[meg_]

「やはり同じ結果になる」の意味が分かりませんが、当方では回答に追記したコードで動作確認出来ました。