要將 numpy 數組從 bool 轉換為 int，我應該乘以 1 還是使用 astype(int)？

Question

bool_array * 1和bool_array.astype(int)似乎做同樣的事情。 哪個更有效率？

Answer 1

bool_array.astype(int)更快（ Colab 筆記本）：

import numpy as np

bool_array = np.random.choice([0, 1], size=1_000_000).astype(bool)

%timeit int_array = bool_array * 1

1000 次循環，5 次中的最佳：每個循環 1.5 毫秒

%timeit int_array = bool_array.astype("int")

1000 個循環，5 個循環中的最佳：每個循環 973 µs