여러 Pandas DataFrame 열에 함수 요소별로 어떻게 적용할 수 있나요?

Pandas: 여러 열에 함수 적용

문제:

여러 열이 있는 함수와 Pandas DataFrame이 제공됩니다. 어떻게 함수를 요소별로 적용하여 해당 값의 각 쌍에 적용할 수 있습니까? 열이요?

해결책:

Pandas의 람다 함수와 적용 방법을 사용하여 이를 달성하는 효율적인 방법이 있습니다.

def get_sublist(sta, end):
    return mylist[sta:end+1]

df['col_3'] = df.apply(lambda x: get_sublist(x.col_1, x.col_2), axis=1)

이것이 적용됩니다. 'col_1' 및 'col_2' 열의 값을 입력으로 사용하여 DataFrame의 각 행에 get_sublist 함수를 사용합니다. 결과는 계산된 하위 목록을 포함하는 새 열 'col_3'입니다.

사용법:

다음 예를 고려하세요.

df = pd.DataFrame({'ID':['1','2','3'], 'col_1': [0,2,3], 'col_2':[1,4,5]})
mylist = ['a','b','c','d','e','f']

# Apply the function and create a new column
df['col_3'] = df.apply(lambda x: get_sublist(x.col_1, x.col_2), axis=1)

print(df)

출력:

  ID  col_1  col_2            col_3
0  1      0      1       ['a', 'b']
1  2      2      4  ['c', 'd', 'e']
2  3      3      5  ['d', 'e', 'f']

위 내용은 여러 Pandas DataFrame 열에 함수 요소별로 어떻게 적용할 수 있나요?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!