gpt4 book ai didi

python - 将两个列表按元素连接成一个 n x n 矩阵

转载 作者:行者123 更新时间:2023-12-02 09:50:27 25 4
gpt4 key购买 nike

我有两个带有字符串的列表,我想将它们按元素连接成一个 n x n 矩阵。我已经尝试了下面的代码,但这只给了我 n x 1 列表。

row = ['a','b','c']
col = ['a','b','c']

matrix = map(''.join, zip(row,col))

预期的输出将是这样的矩阵:

[['aa','ab','ac'],
['ba','bb','bc'],
['ca','cb','cc']])

是否有使用常规 python 或 numpy 来完成此任务的解决方案?

最佳答案

常规 Python - 使用字符串连接和列表理解:

matrix = [[x + y for x in row] for y in col]

为了符合 NumPy 样式和所需的输出使用:

np.c_[matrix].T

关于python - 将两个列表按元素连接成一个 n x n 矩阵,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59348226/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com