从两个匹配列表中随机选择[Python]

时间:2018-10-10 10:35:03

标签: python json list scripting devops

从我的代码执行此方案时遇到问题。我必须在.ini文件中列出。

[FMI]
vendorCodes=["a", "b", "c", "d", "e"]
supplierName=["Test 1", "Test 2", "Test 3", "Test 4", "Test 5" ]

我使用json.loads从.ini文件中调用列表。

vendorCodes = json.loads(config.get('Section1','vendorCodes'))
supplierName = json.loads(config.get('Section1','supplierName'))4

从列表中,random.choice将执行以选择数据。

vendorCodes, supplierName = random.choice(vendorCodes), random.choice(supplierName)

我还有一个将其写入.csv文件中的函数。

fmiwriter.writerow([poNumber] + [vendorCodes] + [supplierName])

问题是输出不正确。它应该与列表匹配,如果vendorCodes中的随机选择为a,则supplierName应该为Test 1

这是下面的错误输出:

588636,a,Test 3
332790,c,Test 2
810055,d,Test 4
655819,a,Test 3
159163,e,Test 1

预期输出应如下所示:

588636,a,Test 1
332790,c,Test 3
810055,d,Test 4
655819,a,Test 1
159163,e,Test 5

2 个答案:

答案 0 :(得分:2)

请勿使用choice。而是选择一个整数索引,然后使用该索引访问两个列表。

import random
index = random.randrange(len(vendorCodes))
chosenVendorCode = vendorCodes[index]
chosenSupplierName = supplierName[index]

答案 1 :(得分:1)

您可以zip random.choice的两个列表中串联选择一对。

chosenVendorCode, chosenSupplierName = random.choice(zip(vendorCodes, supplierName))