Question

我的目标是创建一个将CSV文件作为输入的函数。我希望从该CSV文件创建一个数组，其中第一列中的数据为x坐标，第二列中的数据为y坐标。我希望数据文件中的每一行都将数据添加到数组中。

$query = 'SELECT MAX(t2.amount) as maxAmount
          FROM table2 t2
          JOIN table1 t1
            ON t1.type = t2.type
              AND t1.category = t2.category';
$query = $con->query($query);
$result = $query->fetch_assoc();
echo $result['maxAmount'];

当我返回它时，它只是返回CSV文件的最后一行，而实际上我希望创建的所有数组都返回。我使用过的函数，例如.append（），给我一个错误，说“不能在0维数组上使用追加”。关于如何编辑代码以提供返回数组的任何建议。

Answer 1

作为一种快速的（就更改代码而言）解决方案，请使用生成器：

def generateArray(doc):
    for theData in doc: 
        editDocument = theData.strip().split(",")
        x = splitDocument[0]
        y = splitDocument[1]
        createArray = [x, y]
        yield createArray

arr = np.array(list(generateArray(doc)))
print(arr)

作为更好的解决方案，我建议您检查如何使用np.loadtxt。

将数组追加到较大的数组，然后返回

1 个答案: