根据行中的最大大小，在新列中写入'Big‘、'Medium’、‘Small

import pandas as pd

# 假设数据存储在一个名为data的DataFrame中，其中包含一个名为'size'的列
data = pd.DataFrame({'size': [10, 5, 20, 15, 8]})

# 创建一个新的列'size_category'，初始值为空字符串
data['size_category'] = ''

# 遍历每一行的数据
for index, row in data.iterrows():
    # 找到行中的最大大小
    max_size = max(row['size'])
    
    # 根据最大大小的不同范围，将对应的标签写入新列中
    if max_size > 15:
        data.at[index, 'size_category'] = 'Big'
    elif max_size > 10:
        data.at[index, 'size_category'] = 'Medium'
    else:
        data.at[index, 'size_category'] = 'Small'

# 打印处理后的数据
print(data)

   size size_category
0    10         Small
1     5         Small
2    20           Big
3    15        Medium
4     8         Small