python從list列表中選出一個數和其對應的坐標方法

更新時間：2019年07月20日 10:49:05 作者：Mr-Cat伍可貓

今天小編就為大家分享一篇python從list列表中選出一個數和其對應的坐標方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

例1：給一個列表如下，里面每個元素對應的是x和y的值

a = [[5,2],[6,3],[8,8],[1,3]]

現(xiàn)在要挑出y的值為3對應的x的值，即6和1

import numpy as np

a = [[5,2],[6,3],[8,8],[1,3]] 
#c=np.mat(a),因為只有矩陣（也可以用array）才能用a[0,0]這樣的調用
#表示第一個數的用法而list沒有，故在最后append需要用到
#注意：array也沒有index這樣的用法（只有l(wèi)ist有，此題a已經是list），
#使用index要改成list
c = np.mat(a) 
b = []
for i in range(len(a)):
  try:      #使用try...except...是因為比如3不在第一個[5,2]中會報錯并終止運行
    b_index = a[i].index(3)   #使用index尋找3所在的索引值
  except:    
    pass   #pass表示如果報錯那么跳過并繼續(xù)下一個
  else:   #如果沒有報錯則進行else內容
    b.append(c[i,0])
print(b)

#結果為
[6,1]

或者使用簡單的如下：

a = [[5,2],[6,3],[8,8],[1,3]]
x = [i for i in a if 3 in i]
print(x)

例2：

已知一系列坐標(x,y,z,h)分別對應橫坐標縱坐標z坐標和高，現(xiàn)在要在x-y平面畫出h=0.00221圖（實際上就是等高線圖）

h保留5位小數。

上面只展示了一小部分數據

1，導入數據

import matplotlib.pyplot as plt
import numpy as np

data = np.loadtxt(r'/media/sf_Share/fisher_matrix/myLSPE/data/41+95+150/r_0.01/sigma_F_0.1/sigma_F=0.1/threefre.txt')

2，將第四列即h保留5位小數

for i in range(len(data)):
  data[i,3] = float('%.5f'%data[i,3])

3，將h=0.00211對應的x,y坐標找到并保存到x_mat和y_mat中

x_mat = []
y_mat = []
for i in range(len(data)):
  try:
    data_index = list(data[i]).index(0.00211)
  except:
    pass
  else:
    x_mat.append(data[i,0])
    y_mat.append(data[i,1])

4，畫散點圖

plt.scatter(x_mat,y_mat)
plt.show()

結果如下：

完整代碼：

import matplotlib.pyplot as plt
import numpy as np

data = np.loadtxt(r'/media/sf_Share/fisher_matrix/myLSPE/data/41+95+150/r_0.01/sigma_F_0.1/sigma_F=0.1/threefre.txt')

for i in range(len(data)):
  data[i,3] = float('%.5f'%data[i,3])
  
#data_list = list(data)
x_mat = []
y_mat = []
for i in range(len(data)):
  try:
    data_index = list(data[i]).index(0.00211)
  except:
    pass
  else:
    x_mat.append(data[i,0])
    y_mat.append(data[i,1])

plt.scatter(x_mat,y_mat)
plt.show()

以上這篇python從list列表中選出一個數和其對應的坐標方法就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: