时间戳

首页 > 解决方案 > 循环遍历数据框中的唯一值列表

python - 循环遍历数据框中的唯一值列表

问题描述

如何打印此数据框中的每个资产（['Asset A' 'Asset B' 'Asset C']）？我相信我的语法不正确。

import pandas as pd

df = pd.DataFrame({
    'date': ['2019-01-01','2019-01-01','2019-01-01',
             '2019-02-01','2019-02-01','2019-02-01',
             '2019-03-01','2019-03-01','2019-03-01',
             '2019-04-01','2019-04-01','2019-04-01',
             '2019-05-01','2019-05-01','2019-05-01'],
    'Asset': ['Asset A', 'Asset B', 'Asset C', 'Asset A', 'Asset B', 'Asset C',
                'Asset A', 'Asset B', 'Asset C', 'Asset A', 'Asset B', 'Asset C',
                'Asset A', 'Asset B', 'Asset C',],
    'Monthly Value': [200, 800, 400, 400, 300, 600, 400, 400, 900,
                       300, 600, 400, 200, 100, 200],
    'Indicator': [0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 1, 1, 0, 0]
})

print(df.sort_values(by=['Asset']))

          date    Asset  Monthly Value  Indicator
0   2019-01-01  Asset A            200          0
3   2019-02-01  Asset A            400          0
6   2019-03-01  Asset A            400          0
9   2019-04-01  Asset A            300          0
12  2019-05-01  Asset A            200          1
1   2019-01-01  Asset B            800          0
4   2019-02-01  Asset B            300          0
7   2019-03-01  Asset B            400          1
10  2019-04-01  Asset B            600          0
13  2019-05-01  Asset B            100          0
2   2019-01-01  Asset C            400          0
5   2019-02-01  Asset C            600          0
8   2019-03-01  Asset C            900          0
11  2019-04-01  Asset C            400          1
14  2019-05-01  Asset C            200          0

这个循环有问题

asset = df['Asset'].unique()
for asset in df['Asset']:
    print(asset)

资产 A、B 和 C 的所需输出应如下所示。

          date    Asset  Monthly Value  Indicator
2   2019-01-01  Asset C            400          0
5   2019-02-01  Asset C            600          0
8   2019-03-01  Asset C            900          0
11  2019-04-01  Asset C            400          1
14  2019-05-01  Asset C            200          0

标签： python

解决方案

而不是使用unique你应该使用groupby.

for _, v in df.groupby('Asset'):
    print(v, end='\n\n')

          date    Asset  Monthly Value  Indicator
0   2019-01-01  Asset A            200          0
3   2019-02-01  Asset A            400          0
6   2019-03-01  Asset A            400          0
9   2019-04-01  Asset A            300          0
12  2019-05-01  Asset A            200          1

          date    Asset  Monthly Value  Indicator
1   2019-01-01  Asset B            800          0
4   2019-02-01  Asset B            300          0
7   2019-03-01  Asset B            400          1
10  2019-04-01  Asset B            600          0
13  2019-05-01  Asset B            100          0

          date    Asset  Monthly Value  Indicator
2   2019-01-01  Asset C            400          0
5   2019-02-01  Asset C            600          0
8   2019-03-01  Asset C            900          0
11  2019-04-01  Asset C            400          1
14  2019-05-01  Asset C            200          0

如果您想使用该unique方法，那么这就是您要做的方式

asset = df['Asset'].unique()
for a in asset:
    print(df[df.Asset == a], end='\n\n')

推荐阅读