2017-06-18 76 views
0

我想选择一些数据,这要归功于csv文件中的条件,并将其写入另一个组织不同的anoter csv文件中,然后他们再选择另一个数据并将其写入另一个文件中线。我用csv模块和功能writerow做到这一点:在Python中操作csv文件

#Creation of the other csv file 
with open('superAdmins.csv', 'w', newline='') as csvfile: 
    spamwriter = csv.writer(csvfile) 
    spamwriter.writerow(['SuperAdmins'] + ['2 step authentication'] + ['Portée des droits']) 

i = 0 



#Selection of the first data 
with open('admins.csv', 'r') as csvinput,open('2authentication.csv', 'r') as csvinput2, open('superAdmins.csv', 'w+') as csvoutput: 
    reader = csv.DictReader(csvinput, delimiter=',') #lecteur 
    reader2 = csv.DictReader(csvinput2, delimiter=',') #lecteur 

    fieldnames = ['SuperAdmins', '2 step authentication','Portée des droits'] 
    writer = csv.DictWriter(csvoutput,fieldnames = fieldnames) #crayon 

    writer.writeheader() #Ecrit l'en-tête avec les fieldnames 

    for row in reader: 
     # write the selection logic here 
     if row["role"] == "_SEED_ADMIN_ROLE": 
      writer.writerow({'SuperAdmins': row['assignedToUser']}) 


#Selection of the second data 
     for row in reader: 
      print(row["primaryEmail"],superadmin) 
      # write the selection logic here 
      if row["primaryEmail"] == superadmin: 
       if row["isEnforcedIn2Sv"] == "False" and row["isEnrolledIn2Sv"] == "False": 
        writer.writerow({'2 step authentication': "false"}) 

我得到:

SuperAdmins,2 step authentication,Portée des droits 
[email protected],, 
,Test, 
,false, 

而我想有:

SuperAdmins,2 step authentication,Portée des droits 
[email protected],Test,false 

我想这功能writerow通过在每次使用时添加逗号来开始一个新行,我正在寻找一种方法来继续写在同一行上。

非常感谢您的帮助。

回答

0

'writerow'方法写入一整行。所以当你在一行中有多个单元格时,你只需要将整行写一次(而不是现在做的两次)。

简单的解决方案是定义一个变量(也许称为row_dict),用所有的信息逐渐填充它,然后在循环结尾使用writerow函数和变量作为参数。

您的特定代码:

for row in reader: 
    row_dict = {} 
    # write the selection logic here 
    if row["role"] == "_SEED_ADMIN_ROLE": 
     row_dict['SuperAdmins'] = row['assignedToUser'] 

     print(row["primaryEmail"],superadmin) 
     # write the selection logic here 
     if row["primaryEmail"] == superadmin: 
      if row["isEnforcedIn2Sv"] == "False" and row["isEnrolledIn2Sv"] == "False": 
       row_dict['2 step authentication'] = ["false"] 

    writer.writerow(row_dict) 
+0

这应该是一个评论,除非你愿意提供一些代码来的OP质疑 –

+0

谢谢您的回答是相关的。对不起,我开始学习Python,并且从未使用过“row_dict变量”。我不明白网上有关它的信息,我可以问你它是如何工作的吗? – KB303

+0

@DmitryPolonskiy公平点。 – lesingerouge