1. 程式人生 > 其它 >利用Python根據條件輸出.csv檔案的行

利用Python根據條件輸出.csv檔案的行

最近做ChinaVis的比賽,需要對100w的資料進行先處理,這裡我需要根據target的值找出source對應的值輸出出來,即輸出.csv檔案中滿足某一條件的行,故寫了個小demo

csv檔案大致如下:

relation	source	target
r_subdomain	Domain_34a6231f101fdfa2b051beaa4b94d463fe5f9f42b7789bbe60f6fd4c292ee7ac	Domain_5052db3f33d5337ab631025f7d5de3c5ac559edb2c40deda5530c0051f39b1e2

r_dns_a	Domain_34a6231f101fdfa2b051beaa4b94d463fe5f9f42b7789bbe60f6fd4c292ee7ac	IP_37f7ed5739b43757ff23c712ae4d60d16615c59c0818bf5f2c91514c9c695845

r_cert	Domain_34a6231f101fdfa2b051beaa4b94d463fe5f9f42b7789bbe60f6fd4c292ee7ac	Cert_9ace6aae20e3ac6d9ebfae8938b91112460b27ad204cf11f1301f154c5d309a4
...

程式碼實現:

csvfile = open('Link.csv', 'r')
for row in csvfile:
    if str('Cert_6724539e5c0851f37dcf91b7ac85cb35fcd9f8ba4df0107332c308aa53d63bdb') in row:  # str裡為target的值 
        print(row)

這裡也可以再存入一個csv,但過濾後的資料量不大便手動處理了,僅此記錄,有興趣可以試試。