在包含大数据的csv文件上创建用户输入过滤器

时间:2019-04-28 20:17:39

标签: python csv coding-style

我有一个程序可以打开和读取csv格式的文件,其中包含大数据,例如:

State      Crime type Occurrences Year 

CALIFORNIA ROBBERY    12          1999
CALIFORNIA ASSAULT    45          2003
NEW YORK   ARSON      9           1999
CALIFORNIA ARSON      21          2000
TEXAS      THEFT      30          2000
OREGON     ASSAULT    10          2001

我需要通过用户输入创建3个过滤器。例如:

Enter State:
Enter Crime Type:
Enter Year:

如果我输入:

Enter State: CALIFORNIA
Enter Crime: ASSAULT
Enter Year:  2003

Crime Report
State      Crime type Occurrences Year
CALIFORNIA ASSAULT    45          2003

这需要发生。

我不知道如何解决此问题。.我只能以csv格式打开数据并将其读取到Python表中,该表只会打印出每一行。但是,我需要合并搜索过滤器以缩小结果,如上所示。有人熟悉吗?谢谢大家的帮助。

1 个答案:

答案 0 :(得分:0)

Python中的Pandas库允许您查看和处理csv数据。以下解决方案将导入pandas库,并使用table(stack(lapply(test, function(x) unlist(strsplit(as.character(x), ", "))))[2:1]) 函数读取csv并将其加载到数据帧中,然后要求输入值,请注意State和Crime应该是字符串值并转换为{{1} }和Year应该是整数,并强制转换为read_csv(),然后应用简单的查询来从数据框中过滤所需的结果。我们在构建此查询时要牢记所有三个条件都应满足,并且输入字符串也可以小写。

str