时间戳

首页 > 解决方案 > 将符号数组转换为 1 和 0 数组

python - 将符号数组转换为 1 和 0 数组

问题描述

假设我们有一个股票代码数组/字符串：['AMD','AMZN','BABA','FB']。我需要能够将提供的股票代码转换为 1，将其他股票代码转换为 0。

例如，如果我们向上面的数组提供“AMZN”，则结果数组应该看起来：[0,1,0,0]。如果 'FB' 结果应该类似于 [0,0,0,1]。

我需要将其输入人工智能算法。

标签： pythonarraysartificial-intelligence

解决方案

def get_binary_array(input_array, stock_ticker):
    return [1 for thing in input_array if thing == stock_ticker else 0]

推荐阅读