Question

我有一个11×11大小的网格放在散点图上。散点图是100个随机生成的对。在每个网格空间内，是一种分类类型，其中：

A型大于0，但在X和Y轴上小于0.5， B型在X轴和Y轴上都大于0.5，但小于1.5 等...

我想知道每个网格空间中有多少个点，以及该网格空间中存在的对。这部分不是问题，我只是想知道是否有更多pythonic方式来编写我的循环，因为我不想为每个网格空间编写if语句。

我的脚本如下：

    TypeA = []
    TypeB = []

    fig = plt.figure()
    ax = fig.gca()
    ax.set_xticks(np.arange(0.5, 10.5, 1))
    ax.set_yticks(np.arange(0.5, 10.5, 1))

    for ii in range(100):
        RNumX = randint(0, 10)
        RNumY = randint(0, 10)

        print RNumX, RNumY

        hold(True)
        plot1 = plt.scatter(RNumX, RNumY)

        if RNumX >= 0 and RNumX < 0.5:
            if RNumY >= 0 and RNumY < 0.5:
                PairA = (RNumX, RNumY)
                TypeA.append(PairA)

            elif RNumY >= 0.5 and RNumY < 1.5:
                PairB = (RNumX, RNumY)
                TypeB.append(PairB)

    SumA = len(TypeA)
    SumB = len(TypeB)

    print TypeA, SumA
    print TypeB, SumB

    plt.grid()
    plt.show()

Answer 1

您可以键入矩阵并对值进行舍入以查找索引：

from random import random

# An 11 x 11 matrix of lists
Type = 11 * (11 * ([],),)

fig = plt.figure()
ax = fig.gca()
ax.set_xticks(np.arange(0.5, 10.5, 1))
ax.set_yticks(np.arange(0.5, 10.5, 1))

for ii in range(100):
    # If you want to use floats in stead of ints
    RNumX = 0.5 + 10 * random()
    RNumY = 0.5 + 10 * random()

    print RNumX, RNumY

    hold(True)
    plot1 = plt.scatter(RNumX, RNumY)

    # Round the coordinates to find the indices
    Type[int(RNumX + 0.5)][int(RNumY + 0.5)].append((RNumX, RNumY))

# Print all buckets as your snippet implies
for x in Type:
    for y in x:
        print y, len(y)

# Print only buckets with both values in the same range as your question implies
for x in range(11):
    print Type[x][x], len(Type[x][x])

plt.grid()
plt.show()

Answer 2

你可以使用bisect模块来避免那些if语句。这是一个快速而又肮脏的例子。

from functools import total_ordering
import bisect

@total_ordering
class Point:
  def __init__(self, X, Y):
    self.X = X
    self.Y = Y

  def __eq__(self, other):
    return self.X == other.X and self.Y == other.Y

  def __lt__(self, other):
    return self.X < other.X and self.Y < other.Y

print [(float(x)/2, float(x)/2) for  x in xrange(0, 23)]

[（0.0,0.0），（0.5,0.5），（1.0,1.0），（1.5,1.5），（2.0,2.0），（2.5,2.5），（3.0,3.0），（3.5,3.5）），（4.0,4.0），（4.5,4.5），（5.0,5.0），（5.5,5.5），（6.0,6.0），（6.5,6.5），（7.0,7.0），（7.5,7.5），（8.0,8.0），（8.5,8.5），（9.0,9.0），（9.5,9.5），（10.0,10.0），（10.5,10.5），（11.0,11.0）]

points = [Point(float(x)/2, float(x)/2) for  x in xrange(0, 23)]
types = [chr(x) for x in xrange(65, 91)]
print types

['A'，'B'，'C'，'D'，'E'，'F'，'G'，'H'，'I'，'J'，'K'，'L '，'M'，'N'，'O'，'P'，'Q'，'R'，'S'，'T'，'U'，'V'，'W'，'X'， 'Y'，'Z']

print types[bisect.bisect(points, Point(0.1, 0.1)) - 1]

A

print types[bisect.bisect(points, Point(0.6, 0.6)) - 1]

乙

Python：循环网格分类的更多Pythonic方法

2 个答案: