python 3.x – Passing list-likes to .loc or [] with any missing labels

Tem um erro nesta linha class1 = data4clustering.loc(data(data('mortality')==1).index.unique()).mean() do código que segue:

count = 0
for variable in variables:
    count += 1
    data4clustering = timeseries_data(data_median_hour, variable, filldata = 0)
    print('Plotting ' + str(data4clustering.count().sum()) + ' observations from ' + str(data4clustering.shape(0)) + ' ICU stays' + ' - ' + variable)    
    class1 = data4clustering.loc(data(data('mortality')==1).index.unique()).mean()
    class0 = data4clustering.loc(data(data('mortality')==0).index.unique()).mean()
    plt.subplot(4, 3, count)
    plt.plot(class1)
    plt.plot(class0)
    plt.xlabel('Hours since ICU admission')
    plt.ylabel('Mean ' + variable)

fig.tight_layout()
plt.legend(('Non-survivors', 'Survivors'))    
plt.show()

Apresentando esta mensagem de erro:

KeyError: “Passing list-likes to .loc or () with any missing labels is no longer supported…

See https://pandas.pydata.org/pandas-docs/stable/user_guide/indexing.html#deprecate-loc-reindex-listlike”

Entendi que tem algo no “.loc” que está depreciado. Mesmo consultando a documentação (URL acima) e outras respostas nesta comunidade não estou conseguindo adaptar o código porque a máquina reinicia automaticamente e em seguida o Google Colab informa que foi utilizada toda a memória RAM.

Alguém poderia dar alguma dica?