csv 如何获取Pandas Dataframe 中的列值

mspsb9vt  于 2023-01-06  发布在  其他
关注(0)|答案(3)|浏览(108)

| 位置|类别|计数|
| - ------|- ------|- ------|
| 杜利亚詹地区|局部放电|1个|
| 杜利亚詹地区|标准误|三个|
| 哈日|颜色|1个|
| 哈日|其他|1个|
| 卡塔尔古里|颜色|1个|
| 卡塔尔古里|双磷酸-磷酸脱氢酶|1个|
我怎样才能在变量中得到计数呢?(SE - 3),(PD - 1),就像这样

uqjltbpv

uqjltbpv1#

也许您可以使用to_dict

>>> tuple(df.set_index('Category')['Count'].to_dict().items())
(('PD', 1), ('SE', 3), ('COL', 1), ('OTH', 1), ('DP-PD', 1))

如果您的输出必须是一个元组,并且您想提取一个键,则可以用途:

count = tuple(df.set_index('Category')['Count'].to_dict().items())
print(dict(count)['PD'])

# Output
1
8iwquhpp

8iwquhpp2#

# import pandas library
import pandas as pd

# dictionary
dict = {'Location': ['Duliajan Area', 'Duliajan Area',
            'HAPJAN', 'HAPJAN'],
        'Category': ['PD', 'SE', 
              'COL', 'OTH']
         'Count': [1, 3,
                   1, 1]}

# create a dataframe object
df = pd.DataFrame(dict)

# show the dataframe
print(df)

# list of values of 'category' column
category_list = df['Category'].tolist()

# show the list
print(category_list)
rryofs0p

rryofs0p3#

为此,您可以使用panda库,也可以使用for循环来循环列的长度,您可以使用.shape方法来找到长度。要获得(SE - 3),(PD - 1),您可以使用f"({category_column[i]}-{count_column[i]})"
所以你的最终代码应该是:进口Pandas当PD

df = pd.read_csv("file3.csv")
category_column = df[" Category"]
count_column = df[" Count"]
num_rows = df.shape[0]
new_list = []
for i in range(num_rows):
    new_list.append(f"({category_column[i]}-{count_column[i]})")

相关问题