仅将更改应用于未知边界（墙壁）内的表格

Question

我希望我的循环仅将“墙”内的表格单元格从 0 更改为 5。 “墙”是用户定义的，可以是任何形状，基于坐标。 该图仅用于可视化。

import matplotlib.pyplot as plt 
import pandas as pd

wallPointsX = [5,5,30,30,55,55,5]
wallPointsY = [5,30,30,55,55,5,5]

df = pd.DataFrame(0, index=range(60), columns=range(60))

for x in range(0, 60):
    for y in range(0, 60):
        df[x][y] = 5 #Should only apply inside "walls"

plt.plot(wallPointsX, wallPointsY)
plt.pcolor(df)
plt.show()

Result plot

Answer 1

好的，花了我一些时间，但这样做很有趣。这里的想法是首先从定义墙壁的坐标中创建一个连续的path。接下来，创建一个对象Path。现在循环遍历 DataFrame 中的每个点，然后使用 contains_point 查看创建的 Path 是否包含该 (x,y) 点。我还必须在if 语句中使用条件x==55 and (5<y<=55)，以便包含与 最右边的墙。

import matplotlib.path as mplPath
import numpy as np

wallPointsX = [5,5,30,30,55,55,5]
wallPointsY = [5,30,30,55,55,5,5]

# Create a continuous path across the wall coordinates
path = np.array([list(i) for i in zip(wallPointsX, wallPointsY)])
Path = mplPath.Path(path)

df = pd.DataFrame(0, index=range(60), columns=range(60))

for x in range(0, 60):
    for y in range(0, 60):
        if Path.contains_point((x,y)) or (x==55 and (5<y<=55)):
            df[x-1][y-1] = 5 #Should only apply inside "walls"

plt.plot(wallPointsX, wallPointsY)
plt.pcolor(df)

输出