Closed
Description
df = pd.DataFrame({"A": [1, 2, 3], "B": [4, 5, 6]})
arr = pd.array([7, pd.NA, 9])
mask = np.ones(df.shape, dtype=bool)
>>> result = df.where(mask, arr)
AttributeError: 'IntegerArray' object has no attribute 'reshape'
we can kludge around this by using np.reshape(arr, shape)
instead of arr.reshape(shape)
, but that'll cast to object, which is not great.
Non-kludge solution: 2D EAs