在for循环中添加Pandas列值？

Question

I have the following Python code (not 100% complete):

labels = []
for i in range(0, len(test)):
    name = test.iloc[i]['name']
    s = name.split('.')
    label_name = s[0]
    label = label_names.index(label_name)
    id = int(s[1])
    labels.append(label)
test['label'] = labels

The last statement generates a "A value is trying to be set on a copy of a slice from a DataFrame." warning.

Is there a better way to create the new column 'label' in the dataframe?

Answer 1

def my_function(input):
   name = input
   s = name.split('.')
   label_name = s[0]
   label = label_names.index(label_name)
   id = int(s[1])

   return label

test['label'] = test['name'].apply(lambda x: my_function(x))