Python Pandas - Insert list items in order in a column

Question

That’s the kind of doubt that gets hard to even explain, but let’s try.

Follows the code:

data = {"Id": ["01", "02", "03", "04",'05'],"Fruta": ['bananaAAA','bananaBBB','bananaCCC','bananaDDD','bananaEEE']}
base_dados = pd.DataFrame(data)
display(base_Dados)

    Id  Fruta
0   01  bananaAAA
1   02  bananaBBB
2   03  bananaCCC
3   04  bananaDDD
4   05  bananaEEE

This is the dataframe, as you can see, after all "banana" has a type of code, for example: AAA, BBB, CCC... Well, I’ve only taken those points from those items in the code below.

for fruta in base_dados['Fruta']:
    fruta = fruta[6:9]
    print(fruta)

AAA
BBB
CCC
DDD
EEE

So far so good, but I would like to insert a new column in the dataframe(base_data) with this string information, as in the code below

base_dados.insert(2, 'tipo', fruta)
display(base_dados)
    Id  Fruta   tipo
0   01  bananaAAA   EEE
1   02  bananaBBB   EEE
2   03  bananaCCC   EEE
3   04  bananaDDD   EEE
4   05  bananaEEE   EEE

However, as seen, items are not inserted in order. Here’s an example of how I wanted:

    Id  Fruta   tipo
0   01  bananaAAA   AAA
1   02  bananaBBB   BBB
2   03  bananaCCC   CCC
3   04  bananaDDD   DDD
4   05  bananaEEE   EEE

Answer 1

To do this you must create another column in the datraframe and use the method .apply() to apply a formula for each line.

The code to create the "type" column looks like this:

base_dados['tipo'] = base_dados['Fruta'].apply(lambda x: x[6:9])

Upshot:

	Id	Fruit	guy
0	01	bananaAAA	AAA
1	02	bananaBBB	BBB
2	03	bananaCCC	CCC
3	04	bananaDDD	DDD
4	05	bananaEE	EEE

Your mistake happened because you were resetting the variable fruta and applying again to all columns.