python从list列表中选出一个数和其对应的坐标方法

yipeiwu_com6年前Python基础

例1:给一个列表如下,里面每个元素对应的是x和y的值

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

现在要挑出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这样的用法(只有list有,此题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表示如果报错那么跳过并继续下一个
  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坐标和高,现在要在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列表中选出一个数和其对应的坐标方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持【听图阁-专注于Python设计】。

相关文章

使用PyQt4 设置TextEdit背景的方法

先简单做个分享,有时间再补一个完整的用例 self.textEdit = QtGui.QTextEdit(slef) 1.设置背景颜色 self.textEdit.setSty...

浅谈Python实现Apriori算法介绍

浅谈Python实现Apriori算法介绍

导读: 随着大数据概念的火热,啤酒与尿布的故事广为人知。我们如何发现买啤酒的人往往也会买尿布这一规律?数据挖掘中的用于挖掘频繁项集和关联规则的Apriori算法可以告诉我们。本文首先对A...

利用Python和OpenCV库将URL转换为OpenCV格式的方法

利用Python和OpenCV库将URL转换为OpenCV格式的方法

今天的博客是直接来源于我自己的个人工具函数库。 过去几个月,有些PyImageSearch读者电邮问我:“如何获取URL指向的图片并将其转换成OpenCV格式(不用将其写入磁盘再读...

python图片验证码生成代码

本文实例为大家分享了python图片验证码实现代码,供大家参考,具体内容如下 #!/usr/bin/env python # -*- coding: UTF-8 -*- impo...

python 中文乱码问题深入分析

python 中文乱码问题深入分析

在本文中,以'哈'来解释作示例解释所有的问题,“哈”的各种编码如下: 1. UNICODE (UTF8-16),C854; 2. UTF-8,E59388; 3. GBK,B9FE。 一...