Question

如果列表存储在csv文件下面的示例中，那么每行都存储在数组中吗？

import csv
import os

DIR = "C:/Users/Administrator/Desktop/key_list.csv"

def Customer_List(csv):
    customer = open(DIR)
    for line in customer:
        row = []
        (row['MEM_ID'],
         row['MEM_SQ'],
         row['X_AUTH_USER'],
         row['X_AUTH_KEY'],
         row['X_STORAGE_URL'])=line.split(",")

        if csv == row['MEM_ID']:
            customer.close()
            return(row)
        else:
            print ("Not search for ID")
            return([])
query = input("Input the your email id: ")
result = Customer_List(query)

此示例警告错误代码..为什么..？另外更新此代码＆amp;错误

Input the your email id: sdfsdf@naver.com
Traceback (most recent call last):
  File "C:\Users\Administrator\Desktop\PyDev\Pydev\Day4\uCloudStorage.py", line 32, in <module>
    result = Customer_List(query)
  File "C:\Users\Administrator\Desktop\PyDev\Pydev\Day4\uCloudStorage.py", line 20, in Customer_List
    row['X_STORAGE_URL'])=line.split(",")
ValueError: too many values to unpack (expected 5)

要显示CSV中的内容，请参阅以下简单代码和结果：

DIR = "C:/Users/Administrator/Desktop/key_list.csv"

def Customer_List():
    customer = open(DIR)
    for line in customer:
        print (line)

结果：

MEM_ID, MEM_SQ, X_AUTH_USER, X_AUTH_KEY, X_STORAGE_URL
kimdm98@gmail.com, M100009, M100009:M100009, wreQew3u, AUTH_xxxxxx-xxxxx
minsejisuk@paran.com, M100022, M100022:M100022, PEm6tREx, AUTH_xxxxx-xxxxx
sdfsdf@naver.com, M100034, M100034:M100034, 3tAzEf3u, AUTH_xxxx-xxxxx

=============================================== ============================== 我编辑了这个脚本.....这是最佳实践吗？

DIR = "C:/Users/Administrator/Desktop/key_list.csv"
DATA = csv.reader(open(DIR,"r"))

ID = input("Input the Customer EMAIL ID: ")
def cList(value):
    for row in DATA:
        MEM_ID = row[0]
        MEM_SQ = row[1]
        X_AUTH_USER = row[2]
        X_AUTH_KEY = row[3]
        X_STORAGE_URL = row[4]
        ACCESSKEY = row[5]
        ACCESSKEYID1 = row[6]
        SECRETKEY1 = row[7]
        ACCESSKEYID2 = row[8]
        SECRETKEY2 = row[9]
        if MEM_ID == value:
            print(".EMAIL ID     :" + MEM_ID)
            print(".MID          :" + MEM_SQ)
            print(".PASSWORD     :" + X_AUTH_KEY)
            print(".AUTH_ACCOUNT :" + X_STORAGE_URL)
            print(".API KEY      :" + ACCESSKEY)

cList(ID)
print ("============================")
print ("1. Upload / Download Error")
print ("2. Permission Error")
print ("3. 4xx Error")
print ("4. etc... Error")
print ("============================")

结果

Input the Customer EMAIL ID: kiyoung.jung@kt.com
.EMAIL ID     :kiyoung.jung@kt.com
.MID          :xxxxxx
.PASSWORD     :xxxxxx
.AUTH_ACCOUNT :xxxxxx-d50a-xxxx-xxxbc05-6267d5ff6712
.API KEY      :xxxxxxxx
============================
1. Upload / Download Error
2. Permission Error
3. 4xx Error
4. etc... Error
============================

Answer 1

如果您的输入数据格式与您在问题的最后添加的数据格式相同，那么您可以按照以下方式开始工作：

import csv

DIR = "C:/Users/Administrator/Desktop/key_list.csv"

def Customer_List(email_id):
    with open(DIR, newline='') as f:  # open assuming Python 3.x
        csvreader = csv.reader(f, skipinitialspace=True)
        for fields in csvreader:
            row = {}  # initialize to an empty dictionary
            (row['MEM_ID'],
             row['MEM_SQ'],
             row['X_AUTH_USER'],
             row['X_AUTH_KEY'],
             row['X_STORAGE_URL']) = fields

            if row['MEM_ID'] == email_id:
                return [row['MEM_ID'],
                        row['MEM_SQ'],
                        row['X_AUTH_USER'],
                        row['X_AUTH_KEY'],
                        row['X_STORAGE_URL']]

        else:
            print("ID not found")
            return []

match = Customer_List('minsejisuk@paran.com')
if match:
    print('found! {}'.format(match))

但是你可以通过使用csv.DictReader读取文件来略微简化一些事情，该文件将自动读取标题行以获取字段名，然后使用它们作为每行读取的键返回一个字典：

def Customer_List(email_id):
    with open(DIR, newline='') as f:  # open assuming Python 3.x
        csvreader = csv.DictReader(f, skipinitialspace=True)
        for row in csvreader:
            if row['MEM_ID'] == email_id:
                return [row['MEM_ID'],
                        row['MEM_SQ'],
                        row['X_AUTH_USER'],
                        row['X_AUTH_KEY'],
                        row['X_STORAGE_URL']]
        else:
            print("ID not found")
            return []

将CSV文件中的数据存储到数组中？

1 个答案: