Pandas и работа с csv файлом

import numpy as np 
import pandas as pd 
 
#Функция подсчета по условию задачи
'''
P.s. В столбце genres такой формат, что в одной ячейке несколько жанров, например: Adventure|Fantasy|Action|Western|Thriller, в следующей строке может быть уже меньше жанров, пример: Crime|Thriller
В теории должна разбивать столбец genres на разные жанры, добавлять в массив только новые жанры, а уже потом считать, какой жанр принес наибольшее кол-во прибыли
 
def income(data):
    arr = []
    for i in range(data['profit'].count()):
        a = data.profit[i]
        s = a.find('|')
        b= a[:s]
        if b not in arr[]:
                
    return data
    
'''
 
data = pd.read_csv('films.csv')
 
#Функция feels() и добавление новой колнки profit, где считается прибыль
def feels(data):
    mas = []
    for i in range(data['budget'].count()):
        mas.append(data.revenue[i] - data.budget[i])
    data.insert(2, "profit", mas, True)
    data = data.sort_values(by='profit', ascending=False)
    data = data.drop(data[data['profit'] <= 0].index)
    data = data[['genres', 'profit']]
    return data
 
#Вызов функции
data = feels(data)
#data = income(data)
 
#Вывод
print(data)

import pandas as pd
import numpy as np
df = pd.DataFrame({
    'films': ['one', 'two', 'three', 'four'],
    'genres': ['a|c|d', 'b', 'b|c|d', 'a|c'], 
    'profit': [1, 2, -3, 4],
    'budget': [100, 200, 300, 400]
})
#print(df)
 
 
df['genres'] = df['genres'].map(lambda x: x.split('|'))
df = df.explode('genres').reset_index(drop=True)
#print(df)
 
index = pd.MultiIndex.from_frame(df[['films', 'genres']])
df1 = pd.DataFrame(df['profit'].to_list(), index=index)
df1 = df1.unstack(level=-1).sum(0).reset_index()
df1.drop(df1.columns[[0]], axis=1, inplace=True)
df1.columns = ['genres', 'profit']
#print(df1)
 
print(*df1.loc[df1.profit.argmax(),:].values)

import numpy as np 
import pandas as pd 
 
data = pd.read_csv('films.csv')
 
#Функция подсчета по условию задачи
def income(data):
    data['genres'] = data['genres'].map(lambda x: x.split('|'))
    data = data.explode('genres').reset_index(drop=True)
    index = pd.MultiIndex.from_frame(data[['original_title', 'genres']])
    df1 = pd.DataFrame(data['profit'].to_list(), index=index)
    df1 = df1.unstack(level=-1).sum(0).reset_index()
    df1.drop(df1.columns[[0]], axis=1, inplace=True)
    df1.columns = ['genres', 'profit']
    return df1
 
#Функция feels() и добавление новой колнки profit, где считается прибыль
def feels(data):
    mas = []
    for i in range(data['budget'].count()):
        mas.append(data.revenue[i] - data.budget[i])
    data.insert(2, "profit", mas, True)
    data = data.sort_values(by='profit', ascending=False)
    data = data.drop(data[data['profit'] <= 0].index)
    data = data[['original_title', 'genres', 'profit']]
    return data
 
#Вызов функции
data = feels(data)
data = income(data)
 
#Вывод
print(*data.loc[data.profit.argmax(),:].values)

def income(data):
    data['genres'] = data['genres'].map(lambda x: x.replace(' ', '_').split('|'))
    data = data.explode('genres').reset_index(drop=True)
    df = data.groupby(['genres'])['profit'].agg(['sum']).reset_index()
    df.columns = ['genres', 'profit']
    return df