配置单元，我有一个按时间戳排序的QA数据集(ID、时间、内容、角色)。如何将其转换为(ID，roleA，roleB)格式？

import pandas as pd

# 假设原始数据集为一个名为qa_data的DataFrame，包含四列：ID、时间、内容、角色
# 假设角色有两种：roleA和roleB

# 按时间戳对数据集进行排序
sorted_qa_data = qa_data.sort_values(by='时间')

# 创建新的数据集，包含三列：ID、roleA和roleB
converted_qa_data = pd.DataFrame(columns=['ID', 'roleA', 'roleB'])

# 遍历排序后的数据集
for index, row in sorted_qa_data.iterrows():
    # 提取ID、内容和角色
    qa_id = row['ID']
    content = row['内容']
    role = row['角色']
    
    # 根据角色分配到roleA或roleB列中
    if role == 'roleA':
        converted_qa_data.loc[index] = [qa_id, content, None]
    elif role == 'roleB':
        converted_qa_data.loc[index] = [qa_id, None, content]

# 打印转换后的数据集
print(converted_qa_data)