python读取csv文件，并根据某一列值输出另外一列值

最新推荐文章于 2024-09-14 08:09:55 发布

不愿透露姓名的菜鸟

最新推荐文章于 2024-09-14 08:09:55 发布

阅读量1.1w

点赞数 2

分类专栏： python学习

本文链接：https://blog.csdn.net/Homewm/article/details/84579294

版权

python学习专栏收录该内容

100 篇文章 2 订阅

订阅专栏

csv.DictReader()读取和加载文件，然后根据一列的值处理另外一列的值。

代码很简单，看看就知道啥意思了。

# -*- coding:utf-8 -*-
#author:zgd
import pandas
import csv

with open("url_features_3_zgd.csv", 'r') as f:
    train_csv = csv.DictReader(f)
    firm_type = []
    not_firm_type = []
    for row in train_csv:
        if row["firmware_bool"] == "1":
            if row["getContentType"] not in firm_type:
                firm_type.append(row["getContentType"])
        elif row["firmware_bool"] == "0":
            if row["getContentType"] not in not_firm_type:
                not_firm_type.append(row["getContentType"])
        else:
            print "----------"
    print firm_type
    print not_firm_type