使用pandas将项目追加到空列表中

Question

我正在尝试使用Pandas创建一个经过过滤的数据框，并根据条件使用matplotlib将其显示出来，但是当我尝试将项目添加到一个空列表中时，该列表没有按照条件进行过滤，而是包含了csv中的所有元素。

这是我的代码：

from matplotlib import pyplot as plt
import pandas as pd

plt.style.use('ggplot')
df = pd.read_csv('C:/Users/Gemxs/Documents/Python/Pandas/pokemon.csv')

fig1, (Legendary, Normal) = plt.subplots(nrows=2, ncols=1)

total_pokemon = []
generation_list = []

legendaries = df[df['Legendary'] == True]
total_for_legendaries = df['Total']
generation_for_legendaries = df['Generation']

normals = df[df['Legendary'] == False]
total_for_normals = df[df['HP'] < 500]

for index, row in df.iterrows():
    if row['Total'] > 500:
        total_pokemon.append(row['Total'])

print(total_pokemon)

我想将大于500的元素发送到空列表中。

David Arenburg · Answer

您实际上不需要使用循环，可以直接利用Pandas的tolist()功能来获取系列（Series）作为列表。根据您的需求，可以这样优化代码：

import pandas as pd

df = pd.read_csv('C:/Users/Gemxs/Documents/Python/Pandas/pokemon.csv')

# 创建一个条件掩码，选取'Total'列值大于500的行
condition_mask = df['Total'] > 500

# 应用掩码并直接将筛选出的'Total'列转换为列表
Total = df[condition_mask]['Total'].tolist()

print(Total)

Dmitry Bychenko · Answer

你在使用的循环和条件语句并未正确设置以过滤这些值。让我们修正一下这个问题。

这是你代码的一个修订版：

from matplotlib import pyplot as plt
import pandas as pd

plt.style.use('ggplot')
df = pd.read_csv('C:/Users/Gemxs/Documents/Python/Pandas/pokemon.csv')

# 将数据框根据是否为传说级宝可梦进行分割
legendary_df = df[df['Legendary'] == True]
normal_df = df[df['Legendary'] == False]

# 过滤出总属性值（'Total'）超过500的传说级宝可梦
filtered_legendary = legendary_df[legendary_df['Total'] > 500]

# 从过滤后的数据框中提取'Total'值转为列表
total_values_over_500 = filtered_legendary['Total'].tolist()

print(total_values_over_500)

在这段代码中，我做了以下操作：

将数据框分为两个：一个用于传说级宝可梦（legendary_df），另一个用于普通宝可梦（normal_df）。
从传说级宝可梦的数据框中过滤出那些‘Total’值超过500的。
从这个过滤后的数据框中提取‘Total’列并转为一个列表（total_values_over_500）。

请记住，在使用Pandas处理数据时，通常更高效的做法是利用内置的过滤和列选择方法，而不是通过iterrows()遍历每一行。