我有一个函数 foo 返回形状为 (1000, 2) 的数组
如何将其拆分为两个数组 a(1000) 和 b(1000)
我正在寻找这样的东西:
a;b = foo()
我正在寻找一个可以轻松推广到形状为 (1000, 5) 左右的情况的答案。
The zip(*...)
idiom 转置了传统的多维 Pythonlist
:
x = [[1,2], [3,4], [5,6]]
# get columns
a, b = zip(*x) # zip(*foo())
# a, b = map(list, zip(*x)) # if you prefer lists over tuples
a
# (1, 3, 5)
# get rows
a, b, c = x
a
# [1, 2]
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)