카테고리 없음
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)