如何为csv中的两行创建预定义的键

Question

CSV is of the format

我想制作一本字典，打印后会返回：

Mockito.anyLong()

到目前为止我已经

了

{
'Date': '######'
'Cash': '20000'
'SKY': '5'
'EZJ': '8'
}

我基本上想知道如何仅为前两行创建密钥，并让后两行从CSV中获取密钥。

Answer 1

根据this question及其答案，您可以使用len(row)检索每行令牌的数量。

如果需要，您可以使用另一个变量来跟踪当前行号。

你for循环应如下所示：

int row_number = 0
for row in reader:
    key = ''
    if len(row) > 1:
        key = row[0]
    else:
        key = 'CUSTOM_KEY'  # Depends on the row current row_number

    portfolio[key] = row[1:]
    row_number += 1

备注：这不仅适用于第一行，也适用于少于2个令牌的每一行。如果没有为只有1个标记的所有行指定特定键，则将使用key的默认值（在本例中为空str ''）。