时间戳

首页 > 解决方案 > Python pandas 遍历特定列

python - Python pandas 遍历特定列

问题描述

我正在尝试制作数据集某些特定列的条形图：

for kolom_naam in attributen_dataset:
if kolom_naam in categorische_var:
    print(kolom_naam)
    attributen_dataset.kolom_naam.value_counts().plot(kind='bar')

其中attributen_dataset 是一个大型数据框，categorische_var 是一个包含attributen_dataset 中列名称的列表（列表中的名称是字符串）

我不知道使用“kolom_naam”迭代变量选择列的正确语法。其余的工作，因为 print(kolom_naam) 完全按照它说的做。

谢谢！！！！！

标签： pythonpandasiteration

解决方案

I have tried to do a full mockup using random values. Here I have used categorische_var to filter for column A and C.

Let me know if it works for you:

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np

y = np.random.rand(10,4)
y[:,0]= np.arange(10)



attributen_dataset = pd.DataFrame(y, columns=["X", "A", "B", "C"])

categorische_var=['A', 'C']


for kolom_naam in attributen_dataset:
    if kolom_naam in categorische_var:
        print(kolom_naam)
        #attributen_dataset.kolom_naam.value_counts().plot(kind='bar')

df_new = attributen_dataset[categorische_var]        
df_new.plot(kind="bar")

推荐阅读