使用numpy数组的维数作为if语句python
使用numpy数组的维数作为if语句python
提问于 2018-06-09 00:05:29
我将遍历不同的numpy数组。有些是2D的，有些是1D的。如果评估的数组是1D，我希望将其平铺为2D数组。示例：
c = {'a': np.array([1, 2, 3]), 'b' : np.array([[1, 2, 3], [4, 5, 6]])}
for k in c:
if c[k].shape #is 1D:
c[k] = np.tile(c[k], (len(c[k]),1))我不知道如何处理这种情况。有什么想法吗？我试过像这样的东西
aa = np.array([1, 2, 3])
aa.shape[0]
# 3
aa.shape[1]
# Gives an out of range error我猜可以通过发现没有第二维的数组来发现它是一个一维的数组。但我不知道如何在if语句中对此进行编码。
谢谢
Stack Overflow用户
发布于 2018-06-09 00:11:05
你可以做到
if len(aa.shape)==1:
# something页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/50764517
复制相关文章
相似问题
腾讯云开发者
