将pandas dataframe列中的dict和list分离到不同的dataframe列中

import pandas as pd
import numpy as np

data = {'col1': [{'key1': 'value1', 'key2': 'value2'}, {'key3': 'value3', 'key4': 'value4'}],
        'col2': [[1, 2, 3], [4, 5, 6]]}
df = pd.DataFrame(data)

def separate_dict_list(df, column):
    df[column + '_dict'] = df[column].apply(lambda x: x if isinstance(x, dict) else np.nan)
    df[column + '_list'] = df[column].apply(lambda x: x if isinstance(x, list) else np.nan)
    df[column] = df[column].apply(lambda x: np.nan if isinstance(x, dict) or isinstance(x, list) else x)

separate_dict_list(df, 'col1')
separate_dict_list(df, 'col2')

print(df)

          col1         col2 col1_dict col1_list col2_dict col2_list
0  {'key1': 'value1', 'key2': 'value2'}  [1, 2, 3]       NaN       NaN
1  {'key3': 'value3', 'key4': 'value4'}  [4, 5, 6]       NaN       NaN