Skip to main content

Creating a Plane of best fit from points

Is there a way to make a plane of best fit using matplotlib? I'm trying to get a smooth curved plane, but I'm unsure on how to do so.

My points are arranged as shown in the following image: enter image description here They are quite smooth, except for a few exceptions, which are mostly clear.

My current code is:

from mpl_toolkits import mplot3d
import matplotlib.pyplot as plt
from sklearn import linear_model
plt.style.use('seaborn-poster')

x = np.array([12.5, 10, 9.5, 7.5, 6])
y = np.array([30, 45, 50, 55, 60, 65])
z = np.array([
    [62.13, 55.41, 54.49, 46.46, 42.13],
    [67.11, 59.43, 56.39, 52.64, 41.89],
    [82.89, 61.13, 57.30, 50.75, 43.02],
    [73.31, 60.57, 57.17, 52.64, 41.73],
    [78.11, 62.92, 63.40, 58.08, 48.69],
    [83.96, 65.19, 60.22, 53.57, 44.22]
])


X, Y = np.meshgrid(x, y)
Z = z

x1, y1, z1 = X.flatten(), Y.flatten(), Z.flatten()

X_data = np.array([x1, y1]).reshape((-1, 2))
Y_data = z1

reg = linear_model.LinearRegression().fit(X_data, Y_data)
a1, a2, c = float(reg.coef_[0]), float(reg.coef_[1]), float(reg.intercept_)

fig = plt.figure(figsize = (9,9))
ax = plt.axes(projection='3d')
ax.grid()

ax.plot_surface(X, Y, z)

ax.scatter(X, Y, z, c = 'r', s = 50)
ax.set_title('Figure 1.21 - Plot of Final results')

ax.set_xlabel('Radius of Ball (mm)', labelpad=20)
ax.set_ylabel('Height from which ball was dropped (cm)', labelpad=20)
ax.set_zlabel('Diameter of ripple (mm)', labelpad=20)

plt.show()

I have the a1, a2 and c values using linear regression but how do I plot them? Is linear regression going to give the right sort of result for this graph? I'm quite new to matplotlib, so sorry if this seems obvious.



source https://stackoverflow.com/questions/74910574/creating-a-plane-of-best-fit-from-points

Comments

Popular posts from this blog

Why is my reports service not connecting?

I am trying to pull some data from a Postgres database using Node.js and node-postures but I can't figure out why my service isn't connecting. my routes/index.js file: const express = require('express'); const router = express.Router(); const ordersCountController = require('../controllers/ordersCountController'); const ordersController = require('../controllers/ordersController'); const weeklyReportsController = require('../controllers/weeklyReportsController'); router.get('/orders_count', ordersCountController); router.get('/orders', ordersController); router.get('/weekly_reports', weeklyReportsController); module.exports = router; My controllers/weeklyReportsController.js file: const weeklyReportsService = require('../services/weeklyReportsService'); const weeklyReportsController = async (req, res) => { try { const data = await weeklyReportsService; res.json({data}) console...

How to show number of registered users in Laravel based on usertype?

i'm trying to display data from the database in the admin dashboard i used this: <?php use Illuminate\Support\Facades\DB; $users = DB::table('users')->count(); echo $users; ?> and i have successfully get the correct data from the database but what if i want to display a specific data for example in this user table there is "usertype" that specify if the user is normal user or admin i want to user the same code above but to display a specific usertype i tried this: <?php use Illuminate\Support\Facades\DB; $users = DB::table('users')->count()->WHERE usertype =admin; echo $users; ?> but it didn't work, what am i doing wrong? source https://stackoverflow.com/questions/68199726/how-to-show-number-of-registered-users-in-laravel-based-on-usertype

The number of nodes and edges I calculated manually and by a method provided by networkx library dont match

I am trying to analyse a graph using networkx library in Python. I have implemented a function that returns the nodes and edges of that graph. After that I counted the number of nodes and edges by implementing the len() function for nodes and edges, which gave me 17 for both nodes and edges. However, when I used the method graph.number_of_nodes() of networkx library it gave me 53. My code: import pandas as pd import networkx as nx import matplotlib.pyplot as plt header_list = ["a","b","w"] E =pd.read_csv("aves-barn-swallow-contact-network.edges",sep=" ",header= None,names=header_list) # print(E.head()) # prints first few rows of the csv file G = nx.from_pandas_edgelist(E,"a","b",["w"]) nx.draw(G) plt.show() class GraphAnalyse: #finding nodes and edges def nodeEdge(self,Graph): nodes = Graph.nodes() edges = Graph.edges() ...