(함수 매개 변수로) 열의 이름을 지정할 수있는 파이썬 함수

Question

이 질문에 대한 올바른 검색 용어를 찾을 수 없으므로이 질문에 앞서 질문을 받았으면 사과하십시오.

기본적으로 필자는 특정 종류의 분석을 수행 할 열 (함수 매개 변수로)의 이름을 지정할 수있는 파이썬 함수를 만들고 싶습니다.

예를 들어 아래를 참조하십시오. 분명히이 코드는 작동하지 않습니다. 왜냐하면 'yearattribute'가 df 이후 문자 그대로 취해지기 때문입니다. 도와 줘서 고맙다.

def networkpairs2(df, Year): 
    """ 
    An effort to generalize the networkpairs function by allowing you to choose the 
    organization and actor parameter column names 
    """ 
    totaldf = df 
    yearattribute = '%s' %Year 
    print yearattribute 
    yearlist = list(np.unique(df.yearattribute)) 
    print yearlist 
    return 

Answer 1

내가 당신을 이해한다면, df[yearattribute].unique() 작동합니다 : 아래

는 데모입니다. 사전과 같이 DataFrame 열에 대해 색인을 생성 할 수 있습니다.

제외 # 1 : totaldf = df은 df의 새로운 이름으로 만, 복사본을 만들지 않으며 어쨌든 사용하지 않습니다.

제외 # 2 : 아무 것도 반환하지 않습니다.

Answer 2

현재 getattr를 사용할 수 있습니다

yearlist = list(np.unique(getattr(df, yearattribute))) 

getattr는 그 이름의 캐릭터 라인 표현을 통해 속성에 액세스 할 수 있습니다.

>>> class Foo: 
...  def __init__(self): 
...   self.attr = 'value' 
... 
>>> foo = Foo() 
>>> getattr(foo, 'attr') 
'value' 
>>>