Exemple de programmation Python d'ajout d'une colonne à une matrice numpy

Cet article présente principalement les méthodes de programmation Python pour ajouter une colonne à la matrice numpy. J'espère que cela pourra aider tout le monde.

Nous avons d'abord une matrice numpy dont les données sont un mn. Nous espérons maintenant y ajouter une colonne pour devenir une matrice m(n+1)

<.>

import numpy as np
a = np.array([[1,2,3],[4,5,6],[7,8,9]])
b = np.ones(3)
c = np.array([[1,2,3,1],[4,5,6,1],[7,8,9,1]])
PRint(a)
print(b)
print(c)

[[1 2 3]
 [4 5 6]
 [7 8 9]]
[ 1. 1. 1.]
[[1 2 3 1]
 [4 5 6 1]
 [7 8 9 1]]

Ce que nous devons faire, c'est combiner a et b pour devenir c

Méthode 1

Utilisez np.c_[] et np.r_[] pour ajouter respectivement des lignes et des colonnes

np.c_[a,b]

array([[ 1., 2., 3., 1.],
    [ 4., 5., 6., 1.],
    [ 7., 8., 9., 1.]])

np.c_[a,a]

array([[1, 2, 3, 1, 2, 3],
    [4, 5, 6, 4, 5, 6],
    [7, 8, 9, 7, 8, 9]])

np.c_[b,a]

array([[ 1., 1., 2., 3.],
    [ 1., 4., 5., 6.],
    [ 1., 7., 8., 9.]])

Méthode 2

Utilisez np.insert

np.insert(a, 0, values=b, axis=1)

array([[1, 1, 2, 3],
    [1, 4, 5, 6],
    [1, 7, 8, 9]])

np.insert(a, 3, values=b, axis=1)

array([[1, 2, 3, 1],
    [4, 5, 6, 1],
    [7, 8, 9, 1]])

Méthode 3

Utilisez 'column_stack'

np.column_stack((a,b))

array([[ 1., 2., 3., 1.],
    [ 4., 5., 6., 1.],
    [ 7., 8., 9., 1.]])

Le contenu ci-dessus est un exemple de la façon d'ajouter une colonne à une matrice numpy à l'aide de la programmation Python. J'espère que cela pourra vous aider.

Recommandations associées :

Exemple de démarrage rapide de programmation Python

Explication détaillée de la façon de trier les éléments du dictionnaire dans une liste avec la programmation Python

Introduction à la méthode de programmation Python pour implémenter le tri par fusion

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!