list comprehension for train data_d