python：迭代ya dataframe列作为打印图表的函数中的值

Question

我正在尝试通过数据框中的数字字段迭代，并为Test1创建两个单独的条形图，而另一个则用于按名称分组的Test2分数。我有一个循环，我会遇到类型错误。我有以下数据的少量样本，但是对于大于25个字段的数据框架，循环将运行。以下是我的代码和错误：

import pandas as pd
import matplotlib.pyplot as plt

data = {'Name': ['Tom', 'Joseph', 'Krish', 'John', 'Tom', 'Joseph', 'Krish', 'John'],
                   'Test1': [20, 21, 19, 18, 30, 33, 12, 10],
                   'Test2': [78, 89, 77, 91, 95, 90, 87, 70]}

df = pd.DataFrame(data)

for columns in df.columns[1:]:
    data = df[(df.columns > 80 )].groupby(
        df.Name, as_index = True).agg(
        {columns: "sum"})
    fig, (ax) = plt.subplots( figsize = (24,7))

    data.plot(kind = 'bar', stacked = False,
                  ax = ax)

TypeError: '>' not supported between instances of 'str' and 'int'

Answer 1

您的程序在尝试将“名称”列中的数据与可变定义线路中的整数值进行比较之前，在将其移动到其他两个列之前存在问题。

data = df[(df.columns > 80 )].groupby(df.Name, as_index = True).agg({columns: "sum"})

该列中的值是使函数失败的字符串。通过一些反复试验，我将您的程序修改为仅对第二列和第三列进行比较（“ Test1”和“ Test2”）。以下是修订的代码。

import pandas as pd
import matplotlib.pyplot as plt

data = {'Name': ['Tom', 'Joseph', 'Krish', 'John', 'Tom', 'Joseph', 'Krish', 'John'],
                   'Test1': [20, 21, 19, 18, 30, 33, 12, 10],
                   'Test2': [78, 89, 77, 91, 95, 90, 87, 70]}

df = pd.DataFrame(data)
for columns in df.columns[1:]:
    data = df[(df['Test1'] > 20) | (df['Test2'] > 80)].groupby(df.Name, as_index = True).agg({columns: "sum"})
    fig, (ax) = plt.subplots( figsize = (24,7))
    
    data.plot(kind = 'bar', stacked = False, ax = ax)

plt.show()

运行该程序产生了两个条形图。

您可能需要尝试比较值，但我认为这应该为您提供以下信息的信息。

希望有所帮助。

问候。