아미(아름다운미소)

numba 본문

카테고리 없음

numba

유키공 2025. 6. 12. 10:47
import numpy as np
import pandas as pd
from numba import njit

@njit
def numba_select(condlist, choicelist, default=0):
    output = np.full(condlist[0].shape, default)
    for cond, choice in zip(reversed(condlist), reversed(choicelist)):
        output = np.where(cond, choice, output)
    return output  # NumPy 배열 반환

# 예시 데이터
condlist = [np.array([True, False, False]), np.array([False, True, False])]
choicelist = [np.array([1, 1, 1]), np.array([2, 2, 2])]

# Numba 함수 실행
result_array = numba_select(condlist, choicelist, default=0)

# NumPy 배열 → Pandas DataFrame 변환
df = pd.DataFrame({"result": result_array})
print(df)
Comments