Use variable as functionname and string [duplicate]

Question

I have the following code

import pandas as pd
df = pd.DataFrame(columns=['var1', 'var2','var3'])
df.loc[0] = [0,1,2]

def RS():
    x = 123
    y = 456
    z = 'And some more random shit'
    return x+y

def BS():
    x = -890
    y = (456*1)+90
    z = 'And some more random shit'
    return x-y

def MyCompute(srt, srt_string):
    df[srt_string] = srt()
    df['1min' + srt_string] = 1-df[srt_string]

MyCompute(srt=RS, srt_string='RS')
MyCompute(srt=BS, srt_string='BS')

Is there a way to avoid the double RS and double BS in calling the MyCompute function?

azro · Accepted Answer · 2020-03-02 21:30:06Z

7

Use the attribute __name__ :

def MyCompute(srt):
    df[srt.__name__] = srt()
    df['1min' + srt.__name__] = 1 - df[srt.__name__]

MyCompute(srt=RS)
MyCompute(srt=BS)

answered Mar 2, 2020 at 21:30

azro

54.2k9 gold badges38 silver badges75 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Barmar · Accepted Answer · 2020-03-02 21:29:49Z

1

Put your functions in a dictionary, then you can look up the function by name.

compute_dict = {"RS": RS, "BS": BS}
def MyCompute(srt_string):
    srt = compute_dict[srt_string]
    df[srt_string] = srt()
    df['1min' + srt_string] = 1-df[srt_string]

answered Mar 2, 2020 at 21:29

Barmar

789k57 gold badges554 silver badges669 bronze badges

Comments

wjandrea · Accepted Answer · 2020-03-02 21:29:00Z

0

Yes, you can use the __name__ attribute of a function, e.g. RS.__name__ instead of 'RS'

answered Mar 2, 2020 at 21:29

wjandrea

33.8k10 gold badges69 silver badges105 bronze badges

Comments

quamrana · Accepted Answer · 2020-03-02 21:30:10Z

0

Yes, you can just get the name of the function:

def MyCompute(srt):
    df[srt.__name__] = srt()
    df['1min' + srt.__name__] = 1-df[srt.__name__]

MyCompute(srt=RS)
MyCompute(srt=BS)

answered Mar 2, 2020 at 21:30

quamrana

39.5k13 gold badges57 silver badges77 bronze badges

Collectives™ on Stack Overflow

Use variable as functionname and string [duplicate]

4 Answers 4

Comments

Comments

Comments

Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

Comments

Comments

Comments

Comments

Linked

Related