如何将包含数组中的值的pandas列扩展为多列？

import pandas as pd

# 创建包含数组的列
df = pd.DataFrame({'col': ['A,B,C', 'D,E,F,G', 'H,I']})

# 使用str.split()方法将列拆分为多列
df[['col1', 'col2', 'col3', 'col4']] = df['col'].str.split(',', expand=True)

# 打印结果
print(df)

       col col1 col2 col3 col4
0    A,B,C    A    B    C  NaN
1  D,E,F,G    D    E    F    G
2      H,I    H    I  NaN  NaN