繁体   English   中英

要将 numpy 数组从 bool 转换为 int,我应该乘以 1 还是使用 astype(int)?

[英]To cast a numpy array from bool to int, should I multiply by one or use astype(int)?

bool_array * 1bool_array.astype(int)似乎做同样的事情。 哪个更有效率?

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

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM