如何将（M，N）形状的numpy数组转换为（M，）

Question

例如，我有一个NumPy数组

    A = np.array([
        np.array([1,2]),
        np.array([3,4,7,8]),
        np.array([5,6])
    ], dtype=object)

所以A.shape是(3,)

但是，当第二维的数组大小相同时，

    B = np.array([
        np.array([1,2]),
        np.array([3,4]),
        np.array([5,6])
    ], dtype=object)

然后B.shape是(3,2)。是否可以将B的大小也设为(3,)？

我之所以要使第二个数组成为形状(3,)，是因为我想将列表分配给Pandas DataFrame中的一列，就像这样（df是一个长度为3的DataFrame）

df.loc[:,'B'] = B

但这会导致错误

ValueError: Must have equal len keys and value when setting with an ndarray

（在合同df.loc[:,'A'] = A中，它有效！因此我想使B与A的形状相同）

有人知道如何解决吗？谢谢！