text_plain_output_2.png application_vnd.jupyter.stderr_output_1.png
import pandas as pd

train_df = pd.read_csv('../input/imet-2020-fgvc7/train.csv')
train_df['attribute_ids'] = train_df['attribute_ids'].apply(lambda x: x.split(' '))
train_df['id'] = train_df['id'].apply(lambda x: x + '.png')
print(train_df.shape)
train_df.head()