937135952 发表于 2022-3-4 18:37:12

如何在dataframe里查找某个字段并打出其后跟的值

如题,比如dataframe中,分布着众多城市,城市后一列跟着人口数。但是该dataframe杂乱无章。程序目的只需要实现获取指定城市人口即可。比如想要获取南京的人口6,应该如何编写代码?

阿奇_o 发表于 2022-3-4 23:12:25

本帖最后由 阿奇_o 于 2022-3-4 23:38 编辑

import pandas as pd
import numpy as np
df = pd.read_excel('杂乱的数据.xlsx', header=None)
df.replace('nan', np.nan, inplace=True)
print(df, "\n----------\n")

# 清洗数据:空值处理与提取有效数据,变成"一列"
data =
sd = pd.concat(data).reset_index(drop=True)
print(sd.head(5), "\n----------\n")

# 规整数据:
city, val = [], []
for i, r in enumerate(sd):
    if i%2 == 0:
      city.append(r)
    else:
      val.append(r)

res = pd.DataFrame({'city': city, 'val': val})
print(res)

# res

#%%
用正则表达,更简单,清洗和规整数据这两步,可以一起搞定:
s = str(df.values)
import re
data = re.findall("'(\w+)' (\d+)", s)
print(data)

937135952 发表于 2022-3-5 12:42:54

阿奇_o 发表于 2022-3-4 23:12


非常感谢,还有一个问题想请教,在正则表达式方法中,输出的是[('新疆', '1'), ('上海', '10'), ('北京', '7'), ('杭州', '8'), ('长沙', '2'), ('南京', '6'), ('西安', '7'), ('湖北', '5'), ('海南', '2')]应该怎么print出南京对应的数值呢?

阿奇_o 发表于 2022-3-5 14:25:58

937135952 发表于 2022-3-5 12:42
非常感谢,还有一个问题想请教,在正则表达式方法中,输出的是[('新疆', '1'), ('上海', '10'), ('北京', ...

你这好像在问,"饭菜煮熟了,香喷喷的,可我怎么吃到嘴里呢?"
真可爱~ {:10_304:}

isdkz 发表于 2022-3-5 14:45:16

本帖最后由 isdkz 于 2022-3-6 09:10 编辑

937135952 发表于 2022-3-5 12:42
非常感谢,还有一个问题想请教,在正则表达式方法中,输出的是[('新疆', '1'), ('上海', '10'), ('北京', ...

import pandas as pd
import re
df = pd.read_excel('人口登记表.xlsx', header=None)

pattern = re.compile(r"'([\u4e00-\u9fa5]+)' (\d+)")
string = str(df.values)
while True:
    search = input("请输入你要查找的城市(输入e退出):")
    if search.lower() == 'e':
      break
    try:
      result = next(filter(lambda x: x == search, pattern.finditer(string)))
    except StopIteration:
      print(f"{search} 没有数据!")
      continue
    print(f'{search} 的人口数为 {result}')
页: [1]
查看完整版本: 如何在dataframe里查找某个字段并打出其后跟的值