gpt4 book ai didi

python - numpy:如何根据一堆标准选择行

转载 作者:太空狗 更新时间:2023-10-30 00:26:35 25 4
gpt4 key购买 nike

如何获取第二列等于 4 或 6 的行?

a = np.array(np.mat('1 2; 3 4; 5 6; 7 4'))
b = [4,6]

显然,这不起作用:

c = a[a[:,1] in b]

最佳答案

这样做的 numpythonic 方法是使用 in1d ,类似于:

a[np.in1d(a[:, 1], b)]

关于python - numpy:如何根据一堆标准选择行,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/19111450/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com