确定Pandas df列是否为数组

import pandas as pd
import numpy as np

# 创建一个示例DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [[1, 2], [3, 4], [5, 6]],
    'C': np.array([[1, 2], [3, 4], [5, 6]])
})

# 检查列是否为数组
def is_array(series):
    return series.apply(lambda x: isinstance(x, np.ndarray)).all()

print(is_array(df['A']))  # False
print(is_array(df['B']))  # False
print(is_array(df['C']))  # True

# 检查列的数据类型
print(df.dtypes)

# 判断是否为NumPy数组
def is_numpy_array(series):
    return series.dtype == np.object_ and all(isinstance(x, np.ndarray) for x in series)

print(is_numpy_array(df['A']))  # False
print(is_numpy_array(df['B']))  # False
print(is_numpy_array(df['C']))  # True

# 示例：清洗数据
df['C'] = df['C'].apply(lambda x: x if isinstance(x, np.ndarray) else np.array([x]))

# 示例：使用向量化操作
df['C'].apply(lambda x: x.sum())