如何使用python从这个矩阵中删除行/列

2024-09-28 05:24:07 发布

您现在位置:Python中文网/ 问答频道 /正文

我的矩阵是这样的。

 ['Hotel', ' "excellent"', ' "very good"', ' "average"', ' "poor"', ' "terrible"', ' "cheapest"', ' "rank"', ' "total reviews"']
 ['westin', ' 390', ' 291', ' 70', ' 43', ' 19', ' 215', ' 27', ' 813']
 ['ramada', ' 136', ' 67', ' 53', ' 30', ' 24', ' 149', ' 49', ' 310 ']
 ['sutton place', '489', ' 293', ' 106', ' 39', ' 20', ' 299', ' 24', ' 947']
 ['loden', ' 681', ' 134', ' 17', ' 5', ' 0', ' 199', ' 4', ' 837']
 ['hampton inn downtown', ' 241', ' 166', ' 26', ' 5', ' 1', ' 159', ' 21', ' 439']
 ['shangri la', ' 332', ' 45', ' 20', ' 8', ' 2', ' 325', ' 8', ' 407']
 ['residence inn marriott', ' 22', ' 15', ' 5', ' 0', ' 0', ' 179', ' 35', ' 42']
 ['pan pacific', ' 475', ' 262', ' 86', ' 29', ' 16', ' 249', ' 15', ' 868']
 ['sheraton wall center', ' 277', ' 346', ' 150', ' 80', ' 26', ' 249', ' 45', ' 879']
 ['westin bayshore', ' 390', ' 291', ' 70', ' 43', ' 19', ' 199', ' 813']

我想从中删除第一行和第0列,并创建一个新矩阵。

我该怎么做?

通常在java或so Id中使用以下代码:

 for (int y; y< matrix[x].length; y++)
     for(int x; x < matrix[Y].length; x++)
      {
        if(x == 0 || y == 0)
         {
           continue
          }
          else
           {
             new_matrix[x][y] = matrix[x][y];
           }


      }

python中是否有这样的方法来迭代和有选择地复制元素?

谢谢


编辑

当我在矩阵上迭代时,我还试图将每个矩阵元素从字符串转换为浮点。

这是我根据下面的答案修改的代码。

A = []
f = open("csv_test.csv",'rt')
try:
    reader = csv.reader(f)
    for row in reader:
        A.append(row)
 finally:
     f.close()

 new_list = [row[1:] for row in A[1:]]
 l = np.array(new_list)
 l.astype(np.float32)
 print l

但是我得到一个错误

  --> l.astype(np.float32)
       print l


      ValueError: setting an array element with a sequence.

Tags: csv代码in元素newfornp矩阵
3条回答

如果有列表,则:

new_list = [row[1:] for row in current_list[1:]]

因此,创建一个忽略第一行的新矩阵,对于之后的每一行,忽略第一列。

如果它碰巧是一个numpy.array,那么您可以使用:

your_array[1:,1:]

基本思想

我想到的是:

>>> import numpy as np
>>> l = [['hotel','good','bad'],['hilton',1,2],['ramada',3,4]]
>>> a = np.array(l) # convert to a numpy array to make multi-dimensional slicing possible
>>> a
array([['hotel', 'good', 'bad'],
       ['hilton', '1', '2'],
       ['ramada', '3', '4']], 
      dtype='|S4')
>>> a[1:,1:] # exclude the first row and the first column
array([['1', '2'],
       ['3', '4']], 
      dtype='|S4')
>>> a[1:,1:].astype(np.float32) # convert to float
array([[ 1.,  2.],
       [ 3.,  4.]], dtype=float32)

可以将2d列表传递给numpy数组构造函数,对2d数组进行切片以去掉第一行和第一列,然后使用^{}方法将所有内容转换为浮点。

一句话,那就是:

>>> l = [['hotel','good','bad'],['hilton',1,2],['ramada',3,4]]
>>> np.array(l)[1:,1:].astype(np.float32)
array([[ 1.,  2.],
       [ 3.,  4.]], dtype=float32)

值错误

你得到一个ValueError,因为你实际上有一个锯齿数组。使用问题代码中的变量new_list,您可以向自己证明这一点:

>>> [len(x) for x in new_list]
[9, 9, 9, 9, 9, 9, 9, 9, 9, 8]

最后一行的长度只有8,而不像其他所有行那样是9。给定一个二维锯齿列表,numpy.array构造函数将创建一个1d numpy数组,其dtypeobject。数组中的条目是Python列表。astype调用正试图将Python列表转换为失败的float32。我猜这只是人为失误。如果你把丢失的条目修好了,你就可以走了。

嵌套列表理解是您所需要的。示例:

def remove_from_matrix(matrix, columns, rows):
    return [
           [float(matrix[row_num][col_num])
           for col_num in range(len(matrix[row_num])) 
           if not col_num in columns]

           for row_num in range(len(matrix))
           if not row_num in rows]

相关问题 更多 >

    热门问题